---
datasets:
- FredZhang7/krea-ai-prompts
---

This model is an experimental model by fine tuning 'EleutherAI/gpt-neo-125m' model on Stable Diffusion AI prompts data. 
The code that was used to train the model is found [here](https://github.com/mandar4tech/Text_Gen/blob/main/fine-tune-gpt-neo-for-stable-diffusion-prompt-gen.ipynb) 
This code NB is a reference used from the following [Youtube](https://www.youtube.com/watch?v=uE0_XKh2d6g) video.

@software{gpt-neo,
  author       = {Black, Sid and
                  Gao, Leo and
                  Wang, Phil and
                  Leahy, Connor and
                  Biderman, Stella},
  title        = {{GPT-Neo: Large Scale Autoregressive Language 
                   Modeling with Mesh-Tensorflow}},
  month        = mar,
  year         = 2021,
  note         = {{If you use this software, please cite it using 
                   these metadata.}},
  publisher    = {Zenodo},
  version      = {1.0},
  doi          = {10.5281/zenodo.5297715},
  url          = {https://doi.org/10.5281/zenodo.5297715}
}