eli4s
/

Bert-L12-h384-A6

Inference Endpoints

Model card Files Files and versions Community

eli4s commited on Jul 21, 2021

Commit

3da2e1f

·

1 Parent(s): 3c592ef

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -12,10 +12,10 @@ PS : the tokenizer is the same as the one of the model bert-base-uncased.
 To load the model \& tokenizer :
 ````python
-from transformers import BertForMaskedLM, BertTokenizer
 model_name = "eli4s/Bert-L12-h384-A6"
-model = BertForMaskedLM.from_pretrained(model_name)
 tokenizer = BertTokenizer.from_pretrained(model_name)
 ````

 To load the model \& tokenizer :
 ````python
+from transformers import AutoModelForMaskedLM, BertTokenizer
 model_name = "eli4s/Bert-L12-h384-A6"
+model = AutoModelForMaskedLM.from_pretrained(model_name)
 tokenizer = BertTokenizer.from_pretrained(model_name)
 ````