Nick088 commited on
Commit
016b3c0
1 Parent(s): de0179d

Fixed misspelling "roborovksi/superprompt-v1" to"roborovski/superprompt-v1" in README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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("roborovksi/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")
 
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")