facebook
/

MobileLLM-350M

Text Generation

Model card Files Files and versions Community

zechunliu commited on 23 days ago

Commit

338cbde

•

1 Parent(s): 7831013

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ We are providing 2 ways to run the model:
 To load the pretrained model for further finetuning or evaluation:
 ```bash
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("facebook/MobileLLM-350M", use_fast_tokenizer=False)
 model = AutoModelForCausalLM.from_pretrained("facebook/MobileLLM-350M", trust_remote_code=True)
 ```
 Note that the default tokenizer does not contain special tokens. For example you can use:

 To load the pretrained model for further finetuning or evaluation:
 ```bash
 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("facebook/MobileLLM-125M", use_fast=False)
 model = AutoModelForCausalLM.from_pretrained("facebook/MobileLLM-350M", trust_remote_code=True)
 ```
 Note that the default tokenizer does not contain special tokens. For example you can use: