--- 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](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). - **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