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
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.
Model tree for therealcyberlord/magicwand-gptneo-1.3b
Base model
EleutherAI/gpt-neo-1.3B