facebook
/

MobileLLM-1B

Text Generation

Model card Files Files and versions Community

zechunliu commited on 14 days ago

Commit

f20c8b3

•

1 Parent(s): bd4c88f

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-1B", use_fast_tokenizer=False)
 model = AutoModelForCausalLM.from_pretrained("facebook/MobileLLM-1B", 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-1B", trust_remote_code=True)
 ```
 Note that the default tokenizer does not contain special tokens. For example you can use: