Commit
•
9df1d8d
1
Parent(s):
de0179d
Fixed misspelling "roborovksi/superprompt-v1" to"roborovski/superprompt-v1" in README.md (#3)
Browse files- Fixed misspelling "roborovksi/superprompt-v1" to"roborovski/superprompt-v1" in README.md (016b3c0ae8ff03f18805a2296270655814ca8bef)
Co-authored-by: Nicolas <Nick088@users.noreply.huggingface.co>
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")
|