google
/

flan-t5-small

text2text-generation

text-generation-inference

Model card Files Files and versions

ybelkada commited on Oct 22, 2022

Commit

0394f94

·

1 Parent(s): 99ac30c

Update README.md (#2)

- Update README.md (8332223e6d2e8749732bc5157779bea926d00167)

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -159,7 +159,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
 input_text = "translate English to German: How old are you?"
-input_ids = tokenizer.encode(input_text, return_tensors="pt").input_ids
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))
@@ -180,7 +180,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small", 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")
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))
@@ -204,7 +204,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small", 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")
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))
@@ -225,7 +225,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small", 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")
 outputs = model.generate(input_ids)
 print(tokenizer.decode(outputs[0]))

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