roborovski
/

superprompt-v1

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Use the right tokenizer

#10

by Nick088 - opened May 26

base: refs/heads/main

←

from: refs/pr/10

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ pip install transformers
 ```python
 from transformers import T5Tokenizer, T5ForConditionalGeneration
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto")
 input_text = "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it."

 ```python
 from transformers import T5Tokenizer, T5ForConditionalGeneration
+tokenizer = T5Tokenizer.from_pretrained("roborovski/superprompt-v1")
 model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto")
 input_text = "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it."