Extending stable diffusion prompts with suitable style cues using text generation.
Play with it on HuggingFace Space.
Tokenizer and Model on HuggingFace Hub.
- tokenizer.ipynb: Training a new tokenizer on the dataset of stable diffusion prompts.
- model.ipynb: Training a GPT-2 model on the dataset of stable diffusion prompts using the trained tokenizer.
- inference.ipynb: Trying the prompts generated using the trained model with stable diffusion.
- app.ipynb: Gradio app for the model.
If you want to contribute code, simply create a pull request. If you have an idea, create an issue and the developers will look into it!