Edit model card

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
Downloads last month
0
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for therealcyberlord/magicwand-opt-125m

Base model

facebook/opt-125m
Adapter
(40)
this model

Dataset used to train therealcyberlord/magicwand-opt-125m