Update README.md

README.md

@@ -18,3 +18,18 @@ As such, you can use the following system prompts for different behaviors:
 * A system prompt describing a role-played persona.
 
 For more information, see the Github repository, https://github.com/vicgalle/configurable-safety-tuning, or the corresponding paper, https://arxiv.org/abs/2404.00495
+
+## Citation
+
+If you find this work, data and/or models useful for your research, please consider citing the article:
+
+```
+@misc{gallego2024configurable,
+      title={Configurable Safety Tuning of Language Models with Synthetic Preference Data}, 
+      author={Victor Gallego},
+      year={2024},
+      eprint={2404.00495},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```