Update README.md

README.md

@@ -17,7 +17,7 @@ base_model: stabilityai/stable-diffusion-xl-base-1.0
 <img src="https://github.com/mapo-t2i/mapo/blob/main/assets/mapo_overview.png?raw=true" width=750/>
 </div><br>
 
-We propose **MaPO**, a reference-free, sample-efficient, memory-friendly alignment technique for text-to-image diffusion models. For more details on the technique, please refer to our paper [here] (TODO).
+We propose **MaPO**, a reference-free, sample-efficient, memory-friendly alignment technique for text-to-image diffusion models. For more details on the technique, please refer to our paper [here](https://arxiv.org/abs/2406.06424).
 
 
 ## Developed by
@@ -165,12 +165,12 @@ For qualitative results, please visit our [project website](https://mapo-t2i.git
 ## Citation
 
 ```bibtex
-@misc{todo,
-    title={Margin-aware Preference Optimization for Aligning Diffusion Models without Reference}, 
-    author={Jiwoo Hong and Sayak Paul and Noah Lee and Kashif Rasuland James Thorne and Jongheon Jeong},
-    year={2024},
-    eprint={todo},
-    archivePrefix={arXiv},
-    primaryClass={cs.CV,cs.LG}
+@misc{hong2024marginaware,
+  title={Margin-aware Preference Optimization for Aligning Diffusion Models without Reference}, 
+  author={Jiwoo Hong and Sayak Paul and Noah Lee and Kashif Rasul and James Thorne and Jongheon Jeong},
+  year={2024},
+  eprint={2406.06424},
+  archivePrefix={arXiv},
+  primaryClass={cs.CV}
 }
 ```