Fixed misspelling "roborovksi/superprompt-v1" to"roborovski/superprompt-v1" in README.md
#3
by Nick088 - opened
README.md
CHANGED
|
@@ -17,7 +17,7 @@ pip install transformers
|
|
| 17 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
| 18 |
|
| 19 |
tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
|
| 20 |
-
model = T5ForConditionalGeneration.from_pretrained("
|
| 21 |
|
| 22 |
input_text = "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it."
|
| 23 |
input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
|
|
|
|
| 17 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
| 18 |
|
| 19 |
tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
|
| 20 |
+
model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto")
|
| 21 |
|
| 22 |
input_text = "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it."
|
| 23 |
input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
|