Update README.md

README.md

@@ -26,4 +26,20 @@ Then, after creating a python environment with the correct packages, you can run
 ```
 
 This allows you to directly interact with the model, and shows the diffusion generation process.
-For training or other evaluations, please see our main repository.
+For training or other evaluations, please see our main repository.
+
+## Citation
+
+If you find this work useful, please cite this work as follows.
+
+```bibtex
+@misc{taeivison2025tess2,
+  title={{TESS 2: A Large-Scale Generalist Diffusion Language Model}},
+  author={Jaesung Tae and Hamish Ivison and Sachin Kumar and Arman Cohan},
+  year={2025},
+  eprint={2502.13917},
+  archivePrefix={arXiv},
+  primaryClass={cs.CL},
+  url={https://arxiv.org/abs/2502.13917},
+ }
+```