kuleshov-group
/

mdlm-owt

Model card Files Files and versions Community

subbham commited on Jun 18

Commit

b69bdc9

•

1 Parent(s): 99c700b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ from transformers import AutoModelForMaskedLM, AutoTokenizer
 # See the `Caduceus` collection page on the hub for list of available models.
 tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
-model_name = 'kuleshov-group/mdlm-no_flashattn-fp32-owt'
 model = AutoModelForMaskedLM.from_pretrained(model_name)
 ```

 # See the `Caduceus` collection page on the hub for list of available models.
 tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
+model_name = 'kuleshov-group/mdlm'
 model = AutoModelForMaskedLM.from_pretrained(model_name)
 ```