Views
No views yet
*.experts.*.{gate,up,down}_proj) and dense MLPs (layers.0.mlp.*)*self_attn*), the MoE router (*mlp.gate), the LM head, embeddings, norms, and the MTP layers (*mtp.layers*) are excluded from MXFP4.XiaomiMiMo/MiMo-V2.5-Pro using AMD-Quark in --file2file_quantization mode (per-shard streaming, no full-model materialization). Both weights and activations of the targeted layers are quantized to MXFP4.1cd Quark/examples/torch/language_modeling/llm_ptq/
2exclude_layers="*self_attn* *mlp.gate *mlp.gate.linear *lm_head *embed_tokens* *norm* *mtp.layers*"
3
4python quantize_quark.py \
5 --model_dir /path/to/MiMo-V2.5-Pro \
6 --quant_scheme mxfp4 \
7 --exclude_layers $exclude_layers \
8 --output_dir amd/MiMo-V2.5-Pro-MXFP4 \
9 --model_export hf_format \
10 --file2file_quantization| Benchmark | MiMo-V2.5-Pro (FP8) | MiMo-V2.5-Pro-MXFP4 (this model) | Recovery |
|---|---|---|---|
| GSM8K (flexible-extract) ↑ | 0.9424 | 0.9424 | 100.0% |
| GSM8K (strict-match) ↑ | 0.9416 | 0.9409 | 99.9% |
lm-evaluation-harness framework, with vLLM and amd-quark compiled and installed from source.1lm_eval \
2 --model vllm \
3 --model_args pretrained=amd/MiMo-V2.5-Pro-MXFP4,trust_remote_code=True,tensor_parallel_size=8,dtype=auto,gpu_memory_utilization=0.9,max_model_len=4096,enforce_eager=False \
4 --tasks gsm8k \
5 --num_fewshot 5 \
6 --batch_size autoXiaomiMiMo/MiMo-V2.5-Pro and is released under the same MIT license as the upstream model.