Hugging Face's logo

google
/
t5-xxl-ssm-nqo

Text2Text Generation Transformers PyTorch TensorFlow
English t5 Inference Endpoints text-generation-inference
Model card Files Community
patrickvonplaten commited on
Commit
a3ec36b
•
1 Parent(s): 2a6ea0a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -38,8 +38,8 @@ The model can be used as follows for **closed book question answering**:
38
  ```python
39
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
40
 
41
- t5_qa_model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-large-ssm-nqo")
42
- t5_tok = AutoTokenizer.from_pretrained("google/t5-large-ssm-nqo")
43
 
44
  input_ids = t5_tok("When was Franklin D. Roosevelt born?", return_tensors="pt").input_ids
45
  gen_output = t5_qa_model.generate(input_ids)[0]
38
  ```python
39
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
40
 
41
+ t5_qa_model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-xxl-ssm-nqo")
42
+ t5_tok = AutoTokenizer.from_pretrained("google/t5-xxl-ssm-nqo")
43
 
44
  input_ids = t5_tok("When was Franklin D. Roosevelt born?", return_tensors="pt").input_ids
45
  gen_output = t5_qa_model.generate(input_ids)[0]