Update README.md

README.md

@@ -12,6 +12,7 @@ datasets:
 ---
 <div align="center">
   <h1>Qwen-Image-Pruning</h1>
+    <a href='https://arxiv.org/abs/2511.16156'><img src='https://img.shields.io/badge/arXiv-2508.07607-b31b1b.svg'></a> &nbsp;
 <a href='https://github.com/OPPO-Mente-Lab/Qwen-Image-Pruning'><img src="https://img.shields.io/badge/GitHub-OPPOer-blue.svg?logo=github" alt="GitHub"></a>
 </div>
 
@@ -169,4 +170,22 @@ for path in glob.glob('conds/*'):
             generator=torch.Generator(device="cuda").manual_seed(42),
         ).images[0]
         image.save(os.path.join(output_dir, image_name))
-```
+```
+
+
+
+## Citation
+
+🌟 If you find our work helpful, please consider citing our paper and leaving valuable stars
+
+```
+@misc{ma2025pluggablepruningcontiguouslayer,
+      title={Pluggable Pruning with Contiguous Layer Distillation for Diffusion Transformers}, 
+      author={Jian Ma and Qirong Peng and Xujie Zhu and Peixing Xie and Chen Chen and Haonan Lu},
+      year={2025},
+      eprint={2511.16156},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2511.16156}, 
+}
+```