Gemma4 量化说明

模型介绍

Gemma4 是多模态视觉语言模型,支持图像与文本联合输入。msModelSlim 当前支持 Gemma4 dense 结构的 W8A8 量化最佳实践。

使用前准备

  • 安装 msModelSlim 工具,详情请参见《msModelSlim工具安装指南》

  • 针对 Gemma4,transformers 版本需要 5.5.3:

    pip install "transformers==5.5.3"
    
  • 若使用 --trust_remote_code True,请确保加载的自定义代码文件来源可靠,避免潜在的安全风险。

Gemma4 模型当前已验证的量化方法

模型 原始浮点权重 量化方式 推理框架支持情况 量化命令
gemma-4-31B-it gemma-4-31B-it W8A8量化 vLLM Ascend W8A8量化

说明: 点击量化命令列中的链接可跳转到对应的具体量化命令。

使用示例

gemma-4-31B-it W8A8量化

该模型的量化已集成至一键量化。使用 model_type=gemma-4-31B-itquant_type=w8a8 即可。

msmodelslim quant \
    --model_path /path/to/gemma4_float_weights \
    --save_path /path/to/gemma4_w8a8_weights \
    --device npu \
    --model_type gemma-4-31B-it \
    --quant_type w8a8 \
    --trust_remote_code True

若需指定自定义配置文件,可通过 config_path 指定 gemma4_w8a8.yaml

msmodelslim quant \
    --model_path /path/to/gemma4_float_weights \
    --save_path /path/to/gemma4_w8a8_weights \
    --device npu \
    --model_type gemma-4-31B-it \
    --config_path lab_practice/gemma4/gemma4_w8a8.yaml \
    --trust_remote_code True

附录