Add or Fix Model

README.md

@@ -9,8 +9,8 @@ datasets:
 - bookcorpus
 - wikipedia
 ---
-# MultiBERTs Seed 0 Checkpoint 1500k (uncased)
-Seed 0 intermediate checkpoint 1500k MultiBERTs (pretrained BERT) model on English language using a masked language modeling (MLM) objective. It was introduced in
+# MultiBERTs Seed 1 Checkpoint 20k (uncased)
+Seed 1 intermediate checkpoint 20k MultiBERTs (pretrained BERT) model on English language using a masked language modeling (MLM) objective. It was introduced in
 [this paper](https://arxiv.org/pdf/2106.16163.pdf) and first released in
 [this repository](https://github.com/google-research/language/tree/master/language/multiberts). This model is uncased: it does not make a difference
 between english and English.
@@ -47,7 +47,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
 ```python
 from transformers import BertTokenizer, BertModel
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-model = BertModel.from_pretrained("multiberts-seed-1-1500k")
+model = BertModel.from_pretrained("multiberts-seed-1-20k")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='pt')
 output = model(**encoded_input)