JonasGeiping commited on
Commit
ab7a979
1 Parent(s): 3e6db37

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ import cramming
38
  from transformers import AutoModelForMaskedLM, AutoTokenizer
39
 
40
  tokenizer = AutoTokenizer.from_pretrained("JonasGeiping/crammed-bert")
41
- model = AutoModelForMaskedLM.from_pretrained("JonasGeiping/test-crammedBERT-c5")
42
 
43
  text = "Replace me by any text you'd like."
44
  encoded_input = tokenizer(text, return_tensors='pt')
 
38
  from transformers import AutoModelForMaskedLM, AutoTokenizer
39
 
40
  tokenizer = AutoTokenizer.from_pretrained("JonasGeiping/crammed-bert")
41
+ model = AutoModelForMaskedLM.from_pretrained("JonasGeiping/crammed-bert-legacy")
42
 
43
  text = "Replace me by any text you'd like."
44
  encoded_input = tokenizer(text, return_tensors='pt')