Improve model card: Add pipeline tag, paper link, GitHub & citation

README.md

@@ -1,15 +1,16 @@
 ---
+library_name: transformers
 license: apache-2.0
+pipeline_tag: text-generation
 tags:
 - dllm
 - diffusion
 - llm
-- text_generation
-library_name: transformers
 ---
+
 # LLaDA-MoE
 
-**LLaDA-MoE** is a new and upgraded series of the LLaDA diffusion language model. This pre-release includes two cutting-edge models:
+**LLaDA-MoE** is a new and upgraded series of the LLaDA diffusion language model, developed as part of the `dInfer` framework presented in the paper [dInfer: An Efficient Inference Framework for Diffusion Language Models](https://huggingface.co/papers/2510.08666). This pre-release includes two cutting-edge models:
 
 - `LLaDA-MoE-7B-A1B-Base`: A base pre-trained model designed for research and secondary development.
 - `LLaDA-MoE-7B-A1B-Instruct`: An instruction-tuned model optimized for practical applications.
@@ -20,8 +21,9 @@ library_name: transformers
   <img src="https://raw.githubusercontent.com/Ulov888/LLaDA_Assets/main/benchmarks_details_table.png" width="800" />
 </div>
 
-
-
+## GitHub Repository
+For the complete codebase, training scripts, and more details on the `dInfer` framework, please visit the official GitHub repository:
+[https://github.com/inclusionAI/dInfer](https://github.com/inclusionAI/dInfer)
 
 ## 🚀 Performance Highlights
 
@@ -175,17 +177,20 @@ input_ids = torch.tensor(input_ids).to(device).unsqueeze(0)
 
 text = generate(model, input_ids, steps=128, gen_length=128, block_length=32, temperature=0., cfg_scale=0., remasking='low_confidence')
 print(tokenizer.batch_decode(text[:, input_ids.shape[1]:], skip_special_tokens=False)[0])
-
-
-
-
 ```
 
+## 📚 Citation
 
-## 📚 Citation (Coming Soon)
+If you find `dInfer` and LLaDA-MoE useful in your research or applications, please cite our paper:
 
-We are preparing the technical report and citation information.  
-Stay tuned — citation details will be available soon.
+```bibtex
+@article{dinfer,
+    title={dInfer: An Efficient Inference Framework for Diffusion Language Models},
+    author={Yuxin Ma, Lun Du, Lanning Wei, Kun Chen, Qian Xu, Kangyu Wang, Guofeng Feng, Guoshan Lu, Lin Liu, Xiaojing Qi, Xinyuan Zhang, Zhen Tao, Haibo Feng, Ziyun Jiang, Ying Xu, Zenan Huang, Yihong Zhuang, Haokai Xu, Jiaqi Hu, Zhenzhong Lan, Junbo Zhao, Jianguo Li, Da Zheng},
+    year={2025},
+    journal={arXiv preprint arXiv:2510.08666}
+}
+```
 
 ---
 
@@ -197,6 +202,6 @@ This project is licensed under the terms of the [Apache License 2.0](https://www
 
 ## 🤝 Contact & Collaboration
 
-For questions, collaborations, or feedback, please reach out via [Hugging Face](https://huggingface.co/inclusionAI/LLaDA-MoE-7B-A1B-Base) or open an issue in the [repository](https://github.com/inclusionAI).
+For questions, collaborations, or feedback, please reach out via [Hugging Face](https://huggingface.co/inclusionAI/LLaDA-MoE-7B-A1B-Base) or open an issue in the [repository](https://github.com/inclusionAI/dInfer).
 
 👉 Join us in advancing open, efficient, and intelligent language models!