Update README.md

README.md

@@ -2,4 +2,16 @@
 license: gemma
 ---
 
-A Gemma-2-9B-IT checkpoint implemented with deeper safety alignment, using the data augmentation approach proposed in the paper[Safety Alignment Should Be Made More Than Just a Few Tokens Deep](https://arxiv.org/abs/2406.05946)
+A Gemma-2-9B-IT checkpoint implemented with deeper safety alignment, using the data augmentation approach proposed in the paper [Safety Alignment Should Be Made More Than Just a Few Tokens Deep](https://arxiv.org/abs/2406.05946)
+
+
+
+Citation:
+```
+@article{qi2024safety,
+  title={Safety Alignment Should Be Made More Than Just a Few Tokens Deep},
+  author={Qi, Xiangyu and Panda, Ashwinee and Lyu, Kaifeng and Ma, Xiao and Roy, Subhrajit and Beirami, Ahmad and Mittal, Prateek and Henderson, Peter},
+  journal={arXiv preprint arXiv:2406.05946},
+  year={2024}
+}
+```