Add pipeline tag and GitHub link

README.md

@@ -1,20 +1,26 @@
 ---
-library_name: transformers
-license: mit
 base_model:
 - openai-community/gpt2
+library_name: transformers
+license: mit
+pipeline_tag: text-generation
 ---
+
 # CODI Model
 
 <div align="center">
 
 [![HuggingFace](https://img.shields.io/badge/🤗%20HuggingFace-Model-fcc21b?style=for-the-badge&logo=huggingface&logoColor=white)](https://huggingface.co/dd101bb/latent-tts-codi)
+[![GitHub](https://img.shields.io/badge/GitHub-Code-black?style=for-the-badge&logo=github)](https://github.com/ModalityDance/LatentTTS)
 
 </div>
 
 ## Overview
 
-**CODI** (Continuous Chain-of-Thought via Self-Distillation) is a latent reasoning model based on GPT-2 that extends the base architecture with an optional projector module for enhanced hidden state representations. This model is part of the [Parallel Test-Time Scaling for Latent Reasoning Models](https://arxiv.org/abs/2510.07745) framework.
+**CODI** (Continuous Chain-of-Thought via Self-Distillation) is a latent reasoning model based on GPT-2 that extends the base architecture with an optional projector module for enhanced hidden state representations. This model is part of the [Parallel Test-Time Scaling for Latent Reasoning Models](https://huggingface.co/papers/2510.07745) framework.
+
+- **Paper:** [Parallel Test-Time Scaling for Latent Reasoning Models](https://huggingface.co/papers/2510.07745)
+- **Code:** [Official GitHub Repository](https://github.com/ModalityDance/LatentTTS)
 
 ## Model Details
 
@@ -43,6 +49,8 @@ huggingface-cli download dd101bb/latent-tts-codi --local-dir checkpoints/codi
 
 ### Basic Usage
 
+Note: The following code requires the generation logic provided in the [official repository](https://github.com/ModalityDance/LatentTTS).
+
 ```python
 from transformers import AutoTokenizer
 from src.generation_mixin import LatentGenerationMixin, LatentGenerationConfig
@@ -197,12 +205,6 @@ Run evaluation using the provided scripts:
 ./run_tests.sh
 ```
 
-## Model Card
-
-- **Paper**: [Parallel Test-Time Scaling for Latent Reasoning Models](https://arxiv.org/abs/2510.07745)
-- **HuggingFace**: [dd101bb/latent-tts-codi](https://huggingface.co/dd101bb/latent-tts-codi)
-- **Benchmarks**: GSM8K Test, GSM8K Hard, MultiArith
-
 ## Citation
 
 If you use this model, please cite: