jiaxwang's picture
Update README.md
8eecce0 verified
|
Raw
History Blame
2.77 kB
---
pipeline_tag: image-text-to-text
license: other
license_name: minimax-community
license_link: LICENSE
library_name: transformers
tags:
- multimodal
- moe
- agent
- coding
- video
---
# Model Overview
- **Model Architecture:** MiniMaxM3SparseForConditionalGeneration
- **Input:** Text, Image
- **Output:** Text
- **Supported Hardware Microarchitecture:** AMD MI350/MI355
- **ROCm**: 7.1.1
- **PyTorch**: 2.10.0
- **Transformers**: 5.2.0
- **Operating System(s):** Linux
- **Inference Engine:** [vLLM](https://docs.vllm.ai/en/latest/)
- **Model Optimizer:** [AMD-Quark](https://quark.docs.amd.com/latest/index.html)
- **Weight quantization:** OCP MXFP4, Static; self_attn Perchannel, FP8E4M3, Static
- **Activation quantization:** OCP MXFP4, Dynamic; self_attn Pertoken, FP8E4M3, Dynamic
# Model Quantization
The model was quantized from [MiniMaxAI/MiniMax-M3](https://huggingface.co/MiniMaxAI/MiniMax-M3) using [AMD-Quark](https://quark.docs.amd.com/latest/index.html). The weights are quantized to MXFP4 and activations are quantized to MXFP4, and self_attn layers are quantized to PTPC-FP8.
**Quantization scripts:**
```python
from quark.torch import LLMTemplate, ModelQuantizer
# --- Register template ---
minimax_m3_vl_template = LLMTemplate(
model_type="minimax_m3_vl",
kv_layers_name=["*language_model.*k_proj", "*language_model.*v_proj"],
q_layer_name="*language_model.*q_proj",
exclude_layers_name=[
"*lm_head",
"*vision_tower*",
"*multi_modal_projector*",
"*patch_merge_mlp*",
"*block_sparse_moe.gate",
"*self_attn.index_*",
],
)
LLMTemplate.register_template(minimax_m3_vl_template)
print(f"[INFO]: Registered template '{minimax_m3_vl_template.model_type}'")
# --- Configuration ---
model_dir = "MiniMaxAI/MiniMax-M3"
output_dir = "amd/MiniMax-M3-MXFP4-AttnFP8"
quant_scheme = "mxfp4"
# Per-layer override: self_attn (q/k/v/o_proj) -> ptpc_fp8 instead of mxfp4.
# Equivalent to: --layer_quant_scheme '*self_attn*' ptpc_fp8
layer_config = {
"*self_attn*": "ptpc_fp8",
}
exclude_layers = [
"*lm_head",
"*vision_tower*",
"*multi_modal_projector*",
"*patch_merge_mlp*",
"*block_sparse_moe.gate",
"*self_attn.index_*",
]
# --- Build quant config from template ---
template = LLMTemplate.get("minimax_m3_vl")
quant_config = template.get_config(
scheme=quant_scheme,
layer_config=layer_config,
exclude_layers=exclude_layers,
)
# --- File-to-file quantization (memory-efficient, no full model loading) ---
quantizer = ModelQuantizer(quant_config)
quantizer.direct_quantize_checkpoint(
pretrained_model_path=model_dir,
save_path=output_dir,
)
print(f"[INFO]: Quantization complete. Output saved to {output_dir}")
```