google
/

t5-3b-ssm-nqo

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

patrickvonplaten commited on Dec 7, 2020

Commit

ca5ea4d

•

1 Parent(s): e08d302

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -38,8 +38,8 @@ The model can be used as follows for **closed book question answering**:
 ```python
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
-t5_qa_model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-large-ssm-nqo")
-t5_tok = AutoTokenizer.from_pretrained("google/t5-large-ssm-nqo")
 input_ids = t5_tok("When was Franklin D. Roosevelt born?", return_tensors="pt").input_ids
 gen_output = t5_qa_model.generate(input_ids)[0]

 ```python
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+t5_qa_model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-3b-ssm-nqo")
+t5_tok = AutoTokenizer.from_pretrained("google/t5-3b-ssm-nqo")
 input_ids = t5_tok("When was Franklin D. Roosevelt born?", return_tensors="pt").input_ids
 gen_output = t5_qa_model.generate(input_ids)[0]