davda54 commited on
Commit
0fe403c
1 Parent(s): b81071e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -44,7 +44,7 @@ import torch
44
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
45
 
46
  tokenizer = AutoTokenizer.from_pretrained("ltg/nort5-small")
47
- t5 = AutoModelForSeq2SeqLM.from_pretrained("ltg/nort5-small")
48
 
49
 
50
  # MASKED LANGUAGE MODELING
 
44
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
45
 
46
  tokenizer = AutoTokenizer.from_pretrained("ltg/nort5-small")
47
+ model = AutoModelForSeq2SeqLM.from_pretrained("ltg/nort5-small", trust_remote_code=True)
48
 
49
 
50
  # MASKED LANGUAGE MODELING