Update README.md

README.md

@@ -138,4 +138,20 @@ We train all models using LoRA with the PEFT library. The main parameters are:
 | optim               | paged\_adamw\_32bit |
 | lr\_scheduler\_type |       constant      |
 
-Please check Appendix B of the paper for more details.
+Please check Appendix B of the paper for more details.
+
+# Cite
+
+If you find our work useful, please consider citing it using the following citation:
+
+```
+@misc{puerto2024dcot,
+      title={Fine-Tuning with Divergent Chains of Thought Boosts Reasoning Through Self-Correction in Language Models}, 
+      author={Haritz Puerto and Tilek Chubakov and Xiaodan Zhu and Harish Tayyar Madabushi and Iryna Gurevych},
+      year={2024},
+      eprint={2407.03181},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2407.03181}, 
+}
+```