patrickvonplaten
commited on
Commit
•
ca5ea4d
1
Parent(s):
e08d302
Update README.md
Browse files
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-
|
42 |
-
t5_tok = AutoTokenizer.from_pretrained("google/t5-
|
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-3b-ssm-nqo")
|
42 |
+
t5_tok = AutoTokenizer.from_pretrained("google/t5-3b-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]
|