google
/

t5-xxl-ssm-wq

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

patrickvonplaten commited on Dec 7, 2020

Commit

8855d16

·

1 Parent(s): 263fbf5

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -36,8 +36,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-xxl-ssm-wq")
+t5_tok = AutoTokenizer.from_pretrained("google/t5-xxl-ssm-wq")
 input_ids = t5_tok("When was Franklin D. Roosevelt born?", return_tensors="pt").input_ids
 gen_output = t5_qa_model.generate(input_ids)[0]