bwshen-mi commited on
Commit
2479e2d
Β·
1 Parent(s): d2ba801

Add model card

Browse files
.gitattributes CHANGED
@@ -34,3 +34,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
 
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ assets/architecture.png filter=lfs diff=lfs merge=lfs -text
38
+ assets/mimo-repetition-rate-flash-en.png filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -1,3 +1,219 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ language:
4
+ - en
5
+ - zh
6
+ tags:
7
+ - text-generation
8
+ - multimodal
9
+ - vision-language
10
+ - audio
11
+ - agent
12
+ - video-understanding
13
+ - long-context
14
+ - mimo_v2
15
+ - transformers
16
+ library_name: transformers
17
  ---
18
+
19
+ <br/><br/>
20
+ <div align="center">
21
+ <picture>
22
+ <source srcset="https://github.com/XiaomiMiMo/MiMo/raw/main/figures/Xiaomi_MiMo_darkmode.png?raw=true" media="(prefers-color-scheme: dark)">
23
+ <img src="https://github.com/XiaomiMiMo/MiMo/raw/main/figures/Xiaomi_MiMo.png?raw=true" width="60%" alt="Xiaomi-MiMo" />
24
+ </picture>
25
+ </div>
26
+
27
+ <br/>
28
+
29
+ <div align="center" style="line-height: 1;">
30
+ |
31
+ <a href="https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Flash-MOPD" target="_blank">πŸ€— HuggingFace</a>
32
+ &nbsp;|
33
+ <a href="https://mimo.xiaomi.com/mimo-v2-6" target="_blank">πŸ“° Blog </a>
34
+ &nbsp;|
35
+ <a href="https://platform.xiaomimimo.com" target="_blank">🎨 Xiaomi MiMo API Platform </a>
36
+ &nbsp;|
37
+ <a href="https://aistudio.xiaomimimo.com" target="_blank">πŸ—¨Xiaomi MiMo Studio </a>
38
+ &nbsp;|
39
+ <a href="https://mimo.xiaomimimo.com/desktop/" target="_blank">πŸ’» Xiaomi MiMo Desktop </a>
40
+ &nbsp;|
41
+ </div>
42
+
43
+ <br/>
44
+
45
+ <div align="center" style="line-height: 1.2;">
46
+ <strong>Community</strong><br/>
47
+ <a href="https://huggingface.co/XiaomiMiMo/MiMo-V2.5-Pro/blob/main/assets/wechat.jpg" target="_blank">WeChat Group</a>
48
+ &nbsp;|&nbsp;
49
+ <a href="https://discord.gg/kKC2kNnQEX" target="_blank">Discord</a>
50
+ &nbsp;|&nbsp;
51
+ <a href="https://t.me/+3T-I0pekOVIyNDBl" target="_blank">Telegram</a>
52
+ &nbsp;|&nbsp;
53
+ <a href="https://www.reddit.com/r/XiaomiMiMo_Official/" target="_blank">Reddit</a>
54
+ </div>
55
+
56
+ # MiMo-V2.6-Flash-MOPD
57
+
58
+ <p><a href="https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Flash-RL/blob/main/MiMo_V2_6_technical_report.pdf"><b>Technical Report</b></a></p>
59
+
60
+ > [!IMPORTANT]
61
+ > **This is the MOPD upgrade of the [MiMo-V2.6-Flash-RL](https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Flash-RL) checkpoint.**
62
+ >
63
+ > - **MOPD2** (πŸ‘‰ [**Technical Report Β§5.6**](https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Pro-RL/blob/main/MiMo_V2_6_technical_report.pdf))<br>
64
+ > Fuses several domain-specialized teachers into one model, extending to domains where reliable training-time verification is hard, such as long-horizon game development, scientific research and embodied intelligence.
65
+ > - **Diagnosing and Mitigating Tool-Call Repetition in MiMo-V2.6** (πŸ‘‰ [**Technical Blog**](https://mimo.xiaomi.com/blog/mimo-v2-6-tool-call-repetition))<br>
66
+ > An easy-to-overlook failure mode in which the model keeps issuing the same or highly similar tool calls, appearing busy while making no progress. Nothing fails outright, so it tends to go unnoticed. The MOPD stage handles it efficiently, with a short specialized-teacher run that converges quickly.
67
+
68
+
69
+ ## 1. Introduction
70
+
71
+ ### How MOPD2 works
72
+
73
+ MOPD2 distills several domain-specialized teachers into the student on-policy. The teachers fall into two families: **mixRL teachers**, trained on verifiable tasks, and **SFT teachers**, trained on synthetic demonstrations for open-domain tasks where a reliable reward is hard to design. Three streams contribute to a single update:
74
+
75
+ - **Standard MOPD**: mixRL teachers supervise full autonomous rollouts.
76
+ - **Teacher-Prefix OPD**: prefixes come from teacher rollouts. A trajectory with *k* assistant turns yields *k* history prefixes, one per turn. The model generates a single new turn from each, and the teacher scores it against the same history.
77
+ - **SFT-Prefix OPD**: prefixes come from SFT demonstrations. The demonstration supplies the history, and the model writes its own continuation.
78
+
79
+ Method details are in [Technical Report Β§5.6](https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Pro-RL/blob/main/MiMo_V2_6_technical_report.pdf).
80
+
81
+ ### Tool-call repetition
82
+
83
+ Following the release of MiMo-V2.6, tool-call repetition emerged as one of the most noticeable issues in agentic settings: the model would sometimes issue the same or highly similar tool calls repeatedly, consuming time and context without making progress. This checkpoint mitigates it.
84
+
85
+ ![Tool-call repetition rate on MiMo-V2.6-Flash before and after MOPD](assets/mimo-repetition-rate-flash-en.png)
86
+
87
+ *Figure: response-level repetition rate on MiMo-V2.6-Flash, RL-stage versus this checkpoint, across context lengths and agent harnesses.*
88
+
89
+ The [technical blog](https://mimo.xiaomi.com/blog/mimo-v2-6-tool-call-repetition) has the full diagnosis. The fix is lightweight to train: a short specialized-teacher run that folds into the normal MOPD pass.
90
+
91
+
92
+ ## Model Summary
93
+
94
+ - **Architecture**: Sparse MoE (Mixture of Experts), 309B total / 15B activated parameters
95
+ - **Context Length**: 1M tokens
96
+ - **Modalities**: Text, Image, Video, Audio
97
+ - **Vision Encoder**: 681M-param MiMo ViT (28 layers: 24 SWA + 4 Full)
98
+ - **Audio Encoder**: 308M AudioTokenizer + 127M audio patch encoder
99
+ - **Multi-Token Prediction (MTP)**: 5-layer speculative decoder
100
+
101
+ ![Figure 1: MiMo-V2.6 architecture β€” omni encoders, hybrid SWA backbone, and MTP blocks](assets/architecture.png)
102
+
103
+ *Figure 1. MiMo-V2.6 architecture.*
104
+
105
+ ## 2. Downloads
106
+
107
+ | Model | Download |
108
+ | --- | --- |
109
+ | **MiMo-V2.6-Pro-RL** | [πŸ€— HuggingFace](https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Pro-RL) Β· [πŸ€– ModelScope](https://www.modelscope.cn/models/XiaomiMiMo/MiMo-V2.6-Pro-RL) |
110
+ | **MiMo-V2.6-Flash-RL** | [πŸ€— HuggingFace](https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Flash-RL) Β· [πŸ€– ModelScope](https://www.modelscope.cn/models/XiaomiMiMo/MiMo-V2.6-Flash-RL) |
111
+ | **MiMo-V2.6-Pro-MOPD** | [πŸ€— HuggingFace](https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Pro-MOPD) Β· [πŸ€– ModelScope](https://www.modelscope.cn/models/XiaomiMiMo/MiMo-V2.6-Pro-MOPD) |
112
+ | **MiMo-V2.6-Flash-MOPD** | [πŸ€— HuggingFace](https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Flash-MOPD) Β· [πŸ€– ModelScope](https://www.modelscope.cn/models/XiaomiMiMo/MiMo-V2.6-Flash-MOPD) |
113
+
114
+ ## 3. Model Architecture
115
+
116
+ ### LLM Backbone
117
+
118
+ | Component | MiMo-V2.6-Flash-MOPD |
119
+ | --- | --- |
120
+ | Layers (Total / SWA / GA) | 48 / 39 / 9 |
121
+ | Hidden Size | 4096 |
122
+ | SWA Heads (Q/KV) | 64 / 8 |
123
+ | GA Heads (Q/KV) | 64 / 4 |
124
+ | Head Dimensions (QK / V) | 192 / 128 |
125
+ | Sliding Window Size | 128 |
126
+ | Routed Experts (Total / Activated) | 256 / 8 |
127
+ | Max Context Length | 1M |
128
+ | MTP / Speculative Decoder | 5 SWA layers, window 1024 |
129
+
130
+ The first Transformer block uses global attention with a dense FFN. Remaining blocks interleave local SWA and GA; both use sparse MoE FFNs without shared experts.
131
+
132
+ ### Vision Encoder (MiMo ViT)
133
+
134
+ | Configuration | Value |
135
+ | --- | --- |
136
+ | Layers (Total / SWA / GA) | 28 / 24 / 4 |
137
+ | Hidden Size | 1280 |
138
+ | Attention Heads (Q / KV) | 32 / 8 |
139
+ | Head Dimension | 64 |
140
+ | Patch Size (T Γ— H Γ— W) | 2 Γ— 16 Γ— 16 |
141
+ | Sliding Window (Left / Right) | 64 / 64 |
142
+ | Spatial Merge Size | 2 Γ— 2 |
143
+ | Parameters | 681M |
144
+
145
+ ### Audio Encoders
146
+
147
+ AudioTokenizer encoder: 24 layers (12 SWA / 12 GA), hidden 1024, 20 RVQ codebooks, 308M parameters. Audio patch encoder: 6 layers, 127M parameters; four frames per patch (25 Hz β†’ 6.25 Hz).
148
+
149
+ ### Speculative Decoder
150
+
151
+ 5-layer SWA MTP drafter (DFlash-style). Predicts 7 subsequent tokens per forward pass for parallel verification.
152
+
153
+ ## 4. Deployment
154
+
155
+ For best performance, follow the [SGLang MiMo cookbook](https://docs.sglang.io/cookbook/autoregressive/Xiaomi/MiMo-V2.5). Docker image: `lmsysorg/sglang:latest`.
156
+
157
+ ### SGLang
158
+
159
+ ```bash
160
+ sglang serve \
161
+ --trust-remote-code \
162
+ --model-path XiaomiMiMo/MiMo-V2.6-Flash-MOPD \
163
+ --tp 8 \
164
+ --dp 2 \
165
+ --enable-dp-attention \
166
+ --enable-dp-lm-head \
167
+ --mm-enable-dp-encoder \
168
+ --mem-fraction-static 0.65 \
169
+ --chunked-prefill-size 16384 \
170
+ --speculative-algorithm EAGLE \
171
+ --speculative-num-steps 3 \
172
+ --speculative-eagle-topk 1 \
173
+ --speculative-num-draft-tokens 4 \
174
+ --enable-multi-layer-eagle \
175
+ --reasoning-parser mimo \
176
+ --tool-call-parser mimo \
177
+ --host 0.0.0.0 \
178
+ --port 30000
179
+ ```
180
+
181
+ ### vLLM
182
+
183
+ Follow the [vLLM MiMo-V2.5 recipe](https://recipes.vllm.ai/XiaomiMiMo/MiMo-V2.5). Stable vLLM may lag; pre-built image: `docker pull vllm/vllm-openai:mimov25-cu129`.
184
+
185
+ ```bash
186
+ vllm serve XiaomiMiMo/MiMo-V2.6-Flash-MOPD \
187
+ --tensor-parallel-size 4 \
188
+ --trust-remote-code \
189
+ --gpu-memory-utilization 0.95 \
190
+ --max-model-len auto \
191
+ --reasoning-parser mimo \
192
+ --tool-call-parser mimo \
193
+ --enable-auto-tool-choice \
194
+ --generation-config vllm
195
+ ```
196
+
197
+ Recommended sampling: `temperature=1.0`, `top_p=0.95`.
198
+
199
+ Also available in AI Studio, MiMo Code, Xiaomi MiMo Desktop, Xiaomi MiMo Open Platform API, and OpenRouter.
200
+
201
+ ## Citation
202
+
203
+ ```bibtex
204
+ @misc{mimo2026v26flashmopd,
205
+ title={MiMo-V2.6-Flash-MOPD},
206
+ author={{Xiaomi MiMo Team}},
207
+ year={2026},
208
+ howpublished={\url{https://huggingface.co/XiaomiMiMo/MiMo-V2.6-Flash-MOPD}},
209
+ }
210
+ ```
211
+
212
+ ## Contact
213
+
214
+ For questions or feedback, reach us at [mimo@xiaomi.com](mailto:mimo@xiaomi.com) or join our community:
215
+
216
+ - [WeChat Group](https://work.weixin.qq.com/apph5/external_room/join/group_mng?plg_id=c417f99bd9014b5dd894daa8bfe19790&)
217
+ - [Discord](https://discord.gg/WX2R2uNp)
218
+ - [Telegram](https://t.me/+3T-I0pekOVIyNDBl)
219
+ - [Reddit](https://www.reddit.com/r/XiaomiMiMo_Official/)
assets/architecture.png ADDED

Git LFS Details

  • SHA256: d288768e1771fec19b39ed7dbac4adcbbd2e490384d4ad3c58d259b0c6c6bdcc
  • Pointer size: 131 Bytes
  • Size of remote file: 405 kB
assets/mimo-repetition-rate-flash-en.png ADDED

Git LFS Details

  • SHA256: cf12fd717f3782e9b76c6a2d6285570bf6d58524910fbcc8d22b286990c8a49b
  • Pointer size: 131 Bytes
  • Size of remote file: 477 kB