Commit ·
db17066
1
Parent(s): 55a2adf
Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ We invite you to try out DREAM 💭 for your own application!
|
|
| 31 |
We provide a quick example of how you can try out DREAM with just a few lines of code:
|
| 32 |
```
|
| 33 |
>>> from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 34 |
-
>>> model = AutoModelForSeq2SeqLM.from_pretrained("
|
| 35 |
|
| 36 |
>>> tokenizer = AutoTokenizer.from_pretrained("t5-11b")
|
| 37 |
>>> input_string = "$answer$ ; $question$ = [SITUATION] hitting someones car in the drive thru on purpose. [QUERY] rot"
|
|
|
|
| 31 |
We provide a quick example of how you can try out DREAM with just a few lines of code:
|
| 32 |
```
|
| 33 |
>>> from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 34 |
+
>>> model = AutoModelForSeq2SeqLM.from_pretrained("allenai/DREAM")
|
| 35 |
|
| 36 |
>>> tokenizer = AutoTokenizer.from_pretrained("t5-11b")
|
| 37 |
>>> input_string = "$answer$ ; $question$ = [SITUATION] hitting someones car in the drive thru on purpose. [QUERY] rot"
|