|
--- |
|
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} |
|
} |
|
|
