Update README.md

README.md

@@ -403,13 +403,13 @@ order to determine whether treating anxiety might
 be a potential therapy for improving fog ."""
 
 import torch
-from transformers import LongT5ForConditionalGeneration, AutoTokenizer
+from transformers import AutoTokenizer, LongT5ForConditionalGeneration
 
-tokenizer = LongT5ForConditionalGeneration.from_pretrained("Stancld/longt5-tglobal-large-16384-pubmed-3k_steps")
+tokenizer = AutoTokenizer.from_pretrained("Stancld/longt5-tglobal-large-16384-pubmed-3k_steps")
 
 input_ids = tokenizer(LONG_ARTICLE, return_tensors="pt").input_ids.to("cuda")
 
-model = LEDForConditionalGeneration.from_pretrained("Stancld/longt5-tglobal-large-16384-pubmed-3k_steps", return_dict_in_generate=True).to("cuda")
+model = LongT5ForConditionalGeneration.from_pretrained("Stancld/longt5-tglobal-large-16384-pubmed-3k_steps", return_dict_in_generate=True).to("cuda")
 
 sequences = model.generate(input_ids, global_attention_mask=global_attention_mask).sequences