Create README.md

Create model card

README.md

@@ -0,0 +1,26 @@
+---
+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}
+}