Instructions to use PartAI/TookaBERT-Base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use PartAI/TookaBERT-Base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="PartAI/TookaBERT-Base")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("PartAI/TookaBERT-Base") model = AutoModelForMaskedLM.from_pretrained("PartAI/TookaBERT-Base") - Notebooks
- Google Colab
- Kaggle
| { | |
| "bos_token": "<s>", | |
| "clean_up_tokenization_spaces": true, | |
| "eos_token": "</s>", | |
| "mask_token": "<mask>", | |
| "max_length": null, | |
| "model_input_names": [ | |
| "input_ids", | |
| "attention_mask" | |
| ], | |
| "model_max_length": 512, | |
| "pad_to_multiple_of": null, | |
| "pad_token": "<pad>", | |
| "pad_token_type_id": 0, | |
| "padding_side": "right", | |
| "sep_token": "</s>", | |
| "tokenizer_class": "PreTrainedTokenizerFast", | |
| "truncation_side": "right", | |
| "unk_token": "<unk>" | |
| } | |