Instructions to use roborovski/superprompt-v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use roborovski/superprompt-v1 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("roborovski/superprompt-v1") model = AutoModelForSeq2SeqLM.from_pretrained("roborovski/superprompt-v1", device_map="auto") - Notebooks
- Google Colab
- Kaggle
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")
|