Fill-Mask
Transformers
Safetensors
English
mdlm
custom_code
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -15,9 +15,9 @@ To use the pre-trained model for masked language modeling, use the following sni
15
  from transformers import AutoModelForMaskedLM, AutoTokenizer
16
 
17
  # See the `MDLM` collection page on the hub for list of available models.
18
- tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
19
  model_name = 'kuleshov-group/mdlm-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)
 
15
  from transformers import AutoModelForMaskedLM, AutoTokenizer
16
 
17
  # See the `MDLM` collection page on the hub for list of available models.
18
+ tokenizer = AutoTokenizer.from_pretrained('gpt2')
19
  model_name = 'kuleshov-group/mdlm-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)