library_name: peft
base_model: facebook/opt-125m
pipeline_tag: text-generation
datasets:
- rubend18/DALL-E-Prompts-OpenAI-ChatGPT
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. It is a smaller version of https://huggingface.co./therealcyberlord/magicwand-gptneo-1.3b
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 glowing and floating in a barber shop, realistic
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
config = PeftConfig.from_pretrained("therealcyberlord/magicwand-opt-125m")
model = AutoModelForCausalLM.from_pretrained("facebook/opt-125m")
model = PeftModel.from_pretrained(model, "therealcyberlord/magicwand-opt-125m")
Training Details
Trained for 2000 steps on rubend18/DALL-E-Prompts-OpenAI-ChatGPT
Framework versions
- PEFT 0.7.2.dev0
Evaluation
Testing Data, Factors & Metrics
Testing Data
[More Information Needed]
Factors
[More Information Needed]
Metrics
[More Information Needed]
Results
[More Information Needed]
Summary
Model Examination [optional]
[More Information Needed]
Environmental Impact
Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).
- Hardware Type: [More Information Needed]
- Hours used: [More Information Needed]
- Cloud Provider: [More Information Needed]
- Compute Region: [More Information Needed]
- Carbon Emitted: [More Information Needed]
Technical Specifications [optional]
Model Architecture and Objective
[More Information Needed]
Compute Infrastructure
[More Information Needed]
Hardware
[More Information Needed]
Software
[More Information Needed]
Citation [optional]
BibTeX:
[More Information Needed]
APA:
[More Information Needed]
Glossary [optional]
[More Information Needed]
More Information [optional]
[More Information Needed]
Model Card Authors [optional]
[More Information Needed]
Model Card Contact
[More Information Needed]
Framework versions
- PEFT 0.7.2.dev0