Update README.md

README.md

@@ -158,7 +158,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl")
 
 input_text = "translate English to German: How old are you?"
-input_ids = tokenizer.encode(input_text, return_tensors="pt").input_ids
+input_ids = tokenizer(input_text, return_tensors="pt").input_ids
 
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))
@@ -179,7 +179,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl", device_map="auto")
 
 input_text = "translate English to German: How old are you?"
-input_ids = tokenizer.encode(input_text, return_tensors="pt").input_ids.to("cuda")
+input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
 
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))
@@ -203,7 +203,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl", device_map="auto", torch_dtype=torch.float16)
 
 input_text = "translate English to German: How old are you?"
-input_ids = tokenizer.encode(input_text, return_tensors="pt").input_ids.to("cuda")
+input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
 
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))
@@ -224,7 +224,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl", device_map="auto", load_in_8bit=True)
 
 input_text = "translate English to German: How old are you?"
-input_ids = tokenizer.encode(input_text, return_tensors="pt").input_ids.to("cuda")
+input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
 
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))