Update README.md

README.md

@@ -7,7 +7,7 @@ tags: []
 
 The LION-series are trained using an **empirically optimized pipeline** that consists of three stages: SFT, DPO, and online preference learning (online DPO). We find simple techniques such as sequence packing, loss masking in SFT, increasing the preference dataset size in DPO, and online DPO training can significantly improve the performance of language models. Our best models (the LION-series) **exceed the performance of the official instruct models** tuned with closed-source data and algorithms.
 
-For training datasets, code, and evaluation scripts, please refer to our paper and codebase (to-be-released).
+For training datasets, code, and evaluation scripts, please refer to our [paper](https://arxiv.org/abs/2407.06542) and [codebase](https://github.com/Columbia-NLP-Lab/LionAlignment).
 
 
 ## Model description
@@ -78,15 +78,24 @@ print(prompt)
 
 ### Training details
 
-Please refer to our codebase at (to-be-released).
+Please refer to our [codebase](https://github.com/Columbia-NLP-Lab/LionAlignment).
 
 
-<!-- ## Citation Information
+## Citation Information
 
 If you find this model useful in your work, please consider citing our paper:
+
+```
+@misc{yu2024lionsempiricallyoptimizedapproach,
+      title={LIONs: An Empirically Optimized Approach to Align Language Models}, 
+      author={Xiao Yu and Qingyang Wu and Yu Li and Zhou Yu},
+      year={2024},
+      eprint={2407.06542},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2407.06542}, 
+}
 ```
-@misc{tmp}
-``` -->
 
 ## Acknowledgements