Fixed misspelling "roborovksi/superprompt-v1" to"roborovski/superprompt-v1" in README.md
Browse files
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")
|