Model Description
Magic wand 🪄 is a prompt completion model fine-tuned on text generation models, used for text2img models such as Dalle, Midjourney, and Stable Diffusion.
Uses
Type something that you want to generate and magic wand will complete it for you. Think about it as a friendly guide when you don't have any ideas about what to generate.
Input: Chocolate milk
Completion: Chocolate milk and a cupcake on fire in a rock concert, with a cinematic shot, close-up vintage
This is the output from Dalle-3:
Limitations
Currently the model produces repetitions with longer sequences
How to Get Started with the Model
from peft import PeftModel, PeftConfig
from transformers import AutoModelForCausalLM, AutoTokenizer
config = PeftConfig.from_pretrained("therealcyberlord/magicwand-gptneo-1.3b")
model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
model = PeftModel.from_pretrained(model, "therealcyberlord/magicwand-gptneo-1.3b")
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
Training Details
Trained for 2000 steps on rubend18/DALL-E-Prompts-OpenAI-ChatGPT
Framework versions
- PEFT 0.7.2.dev0
- Downloads last month
- 0
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.
Model tree for therealcyberlord/magicwand-gptneo-1.3b
Base model
EleutherAI/gpt-neo-1.3B