Downloads · 30 days
5
11% of all-time downloads
magicwand-gptneo-1.3b
therealcyberlord/magicwand-gptneo-1.3b
magicwand-gptneo-1.3b is a text generation model from therealcyberlord. Use it when you need the model to write or continue text. It is set up for peft.
Magic wand 🪄 is a prompt completion model fine-tuned on text generation models, used for text2img models such as Dalle, Midjourney, and Stable Diffusion.
All-time downloads
47
Public
Repo size
50.3 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors50.3 MB · 100%
At a glance
- Task
- Text Generation
- Library
- peft
- Access
- Public
- Created
- Jan 8, 2024
- Updated
- Jan 11, 2024
- SHA
- ecdfcce2
Try a prompt
Base models
Model card
From the Hugging Face model README
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
- Task
- Text Generation
- Library
- peft
- Created
- Jan 8, 2024
- Updated
- Jan 11, 2024