gchhablani commited on
Commit
fd8a4ca
1 Parent(s): 888c3c3

Add or Fix Model

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -9,8 +9,8 @@ datasets:
9
  - bookcorpus
10
  - wikipedia
11
  ---
12
- # MultiBERTs Seed 0 Checkpoint 300k (uncased)
13
- Seed 0 intermediate checkpoint 300k MultiBERTs (pretrained BERT) model on English language using a masked language modeling (MLM) objective. It was introduced in
14
  [this paper](https://arxiv.org/pdf/2106.16163.pdf) and first released in
15
  [this repository](https://github.com/google-research/language/tree/master/language/multiberts). This model is uncased: it does not make a difference
16
  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:
47
  ```python
48
  from transformers import BertTokenizer, BertModel
49
  tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
50
- model = BertModel.from_pretrained("multiberts-seed-1-300k")
51
  text = "Replace me by any text you'd like."
52
  encoded_input = tokenizer(text, return_tensors='pt')
53
  output = model(**encoded_input)
 
9
  - bookcorpus
10
  - wikipedia
11
  ---
12
+ # MultiBERTs Seed 1 Checkpoint 20k (uncased)
13
+ Seed 1 intermediate checkpoint 20k MultiBERTs (pretrained BERT) model on English language using a masked language modeling (MLM) objective. It was introduced in
14
  [this paper](https://arxiv.org/pdf/2106.16163.pdf) and first released in
15
  [this repository](https://github.com/google-research/language/tree/master/language/multiberts). This model is uncased: it does not make a difference
16
  between english and English.
 
47
  ```python
48
  from transformers import BertTokenizer, BertModel
49
  tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
50
+ model = BertModel.from_pretrained("multiberts-seed-1-20k")
51
  text = "Replace me by any text you'd like."
52
  encoded_input = tokenizer(text, return_tensors='pt')
53
  output = model(**encoded_input)