Instructions to use SigmarAI/MBART with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SigmarAI/MBART with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("SigmarAI/MBART") model = AutoModelForSeq2SeqLM.from_pretrained("SigmarAI/MBART", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 658 Bytes
f604b1e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | {
"additional_special_tokens": null,
"bos_token": "<s>",
"cls_token": "<s>",
"eos_token": "</s>",
"mask_token": {
"__type": "AddedToken",
"content": "<mask>",
"lstrip": true,
"normalized": true,
"rstrip": false,
"single_word": false
},
"model_max_length": 1024,
"name_or_path": "facebook/mbart-large-50",
"pad_token": "<pad>",
"sep_token": "</s>",
"sp_model_kwargs": {},
"special_tokens_map_file": "/home/suraj/projects/mbart-50/hf_models/mbart-50-large/special_tokens_map.json",
"src_lang": null,
"tgt_lang": null,
"tokenizer_class": "MBartTokenizer",
"tokenizer_file": null,
"unk_token": "<unk>"
}
|