Hugging Face's logo

kuleshov-group
/
mdlm-no_flashattn-fp32-owt

Fill-Mask
Transformers
Safetensors
English
mdlm
custom_code
Model card Files Community
1

Update README.md

#1
by Q-bert - opened
base: refs/heads/main
←
from: refs/pr/1
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -17,7 +17,7 @@ from transformers import AutoModelForMaskedLM, AutoTokenizer
17
  # See the `Caduceus` collection page on the hub for list of available models.
18
  tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
19
  model_name = 'kuleshov-group/mdlm-no_flashattn-fp32-owt'
20
- model = AutoModelForMaskedLM.from_pretrained(model_name)
21
  ```
22
 
23
  For more details, please see our github repository: [MDLM](https://github.com/kuleshov-group/mdlm)
 
17
  # See the `Caduceus` collection page on the hub for list of available models.
18
  tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
19
  model_name = 'kuleshov-group/mdlm-no_flashattn-fp32-owt'
20
+ model = AutoModelForMaskedLM.from_pretrained(model_name, trust_remote_code=True)
21
  ```
22
 
23
  For more details, please see our github repository: [MDLM](https://github.com/kuleshov-group/mdlm)