Sefika commited on
Commit
d61793b
·
verified ·
1 Parent(s): f2a5f01

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ from transformers import AutoModelForCausalLM
35
  from datetime import datetime
36
  from transformers import T5Tokenizer, T5ForConditionalGeneration
37
 
38
- tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
39
  model_id = "Sefika/semeval_base_1"
40
  model = T5ForConditionalGeneration.from_pretrained(model_id,
41
  device_map="auto",
 
35
  from datetime import datetime
36
  from transformers import T5Tokenizer, T5ForConditionalGeneration
37
 
38
+ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-base")
39
  model_id = "Sefika/semeval_base_1"
40
  model = T5ForConditionalGeneration.from_pretrained(model_id,
41
  device_map="auto",