therealcyberlord/magicwand-opt-125m

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