gchhablani commited on
Commit
b2384a8
1 Parent(s): 4b9e051

Update README

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -47,7 +47,7 @@ generation you should look at model like GPT2.
47
  Here is how to use this model to get the features of a given text in PyTorch:
48
  ```python
49
  from transformers import BertTokenizer, BertModel
50
- tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
51
  model = BertModel.from_pretrained("multiberts-seed-0-300k")
52
  text = "Replace me by any text you'd like."
53
  encoded_input = tokenizer(text, return_tensors='pt')
 
47
  Here is how to use this model to get the features of a given text in PyTorch:
48
  ```python
49
  from transformers import BertTokenizer, BertModel
50
+ tokenizer = BertTokenizer.from_pretrained('multiberts-seed-0-300k')
51
  model = BertModel.from_pretrained("multiberts-seed-0-300k")
52
  text = "Replace me by any text you'd like."
53
  encoded_input = tokenizer(text, return_tensors='pt')