therealcyberlord/magicwand-gptneo-1.3b

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