Updated citation info

README.md

@@ -77,6 +77,7 @@ TBD
 
 ### BibTeX entry and citation info
 
+To cite this model, use
 ```bibtex
 @article{gao2020pile,
   title={The Pile: An 800GB Dataset of Diverse Text for Language Modeling},
@@ -85,3 +86,15 @@ TBD
   year={2020}
 }
 ```
+
+To cite the codebase that this model was trained with, use
+
+```bibtex
+@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},
+  url = {http://github.com/eleutherai/gpt-neo},
+  version = {1.0},
+  year = {2021},
+}
+```