Commit
•
64ec016
1
Parent(s):
5499d96
Use the right tokenizer (#10)
Browse files- Use the right tokenizer (8012423706af90471b28cafe989054db9758505c)
Co-authored-by: Nick088 <[email protected]>
README.md
CHANGED
|
@@ -24,7 +24,7 @@ pip install transformers
|
|
| 24 |
```python
|
| 25 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
| 26 |
|
| 27 |
-
tokenizer = T5Tokenizer.from_pretrained("
|
| 28 |
model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto")
|
| 29 |
|
| 30 |
input_text = "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it."
|
|
|
|
| 24 |
```python
|
| 25 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
| 26 |
|
| 27 |
+
tokenizer = T5Tokenizer.from_pretrained("roborovski/superprompt-v1")
|
| 28 |
model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto")
|
| 29 |
|
| 30 |
input_text = "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it."
|