Update README.md

README.md

@@ -12,6 +12,7 @@ datasets:
 ---
 <div align="center">
   <h1>Qwen-Image-Edit-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>
 
@@ -96,4 +97,21 @@ with torch.inference_mode():
     output_image = output.images[0]
     output_image.save("output_image_edit_plus.png")
     print("image saved at", os.path.abspath("output_image_edit_plus.png"))
+```
+
+
+## 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}, 
+}
 ```