Update README.md

README.md

@@ -13,7 +13,15 @@ Please refer to [this repository](https://github.com/MCG-NJU/p-MoD) for our code
 This model is pretrained on [LCS-558K](https://huggingface.co/datasets/liuhaotian/LLaVA-Pretrain) image caption data, and instruction-tuned on [llava-v1_5-mix-665k](https://huggingface.co/datasets/liuhaotian/LLaVA-Instruct-150K/blob/main/llava_v1_5_mix665k.json).
 
 ## Citation
-TBD
+If you find our work helpful for your research and applications, please cite our paper:
+```Bibtex
+@article{zhang2024pmod,
+  title={p-MoD: Building Mixture-of-Depths MLLMs via Progressive Ratio Decay},
+  author={Zhang, Jun and Meng, Desen and Qi, Ji and Huang, Zhenpeng and Wu, Tao and Wang, Limin},
+  journal={arXiv preprint arXiv:2412.04449},
+  year={2024}
+}
+```
 
 ## License
 Llama 2 is licensed under the LLAMA 2 Community License,