Commit
·
9186290
1
Parent(s):
8277318
Update README.md
Browse files
README.md
CHANGED
|
@@ -26,7 +26,7 @@ Authors: *Adam Roberts, Colin Raffel, Noam Shazeer*
|
|
| 26 |
|
| 27 |
|Id | link | Exact Match |
|
| 28 |
|---|---|---|
|
| 29 |
-
|**T5-11b**|**https://huggingface.co/google/t5-
|
| 30 |
|T5-xxl|https://huggingface.co/google/t5-xxl-ssm-wq|43.5|
|
| 31 |
|
| 32 |
## Usage
|
|
@@ -36,8 +36,8 @@ The model can be used as follows for **closed book question answering**:
|
|
| 36 |
```python
|
| 37 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 38 |
|
| 39 |
-
t5_qa_model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-
|
| 40 |
-
t5_tok = AutoTokenizer.from_pretrained("google/t5-
|
| 41 |
|
| 42 |
input_ids = t5_tok("When was Franklin D. Roosevelt born?", return_tensors="pt").input_ids
|
| 43 |
gen_output = t5_qa_model.generate(input_ids)[0]
|
|
|
|
| 26 |
|
| 27 |
|Id | link | Exact Match |
|
| 28 |
|---|---|---|
|
| 29 |
+
|**T5-11b**|**https://huggingface.co/google/t5-11b-ssm-wq**|**44.7**|
|
| 30 |
|T5-xxl|https://huggingface.co/google/t5-xxl-ssm-wq|43.5|
|
| 31 |
|
| 32 |
## Usage
|
|
|
|
| 36 |
```python
|
| 37 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 38 |
|
| 39 |
+
t5_qa_model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-11b-ssm-wq")
|
| 40 |
+
t5_tok = AutoTokenizer.from_pretrained("google/t5-11b-ssm-wq")
|
| 41 |
|
| 42 |
input_ids = t5_tok("When was Franklin D. Roosevelt born?", return_tensors="pt").input_ids
|
| 43 |
gen_output = t5_qa_model.generate(input_ids)[0]
|