llm-compressor 支持
本指南旨在介绍如何使用 LMDeploy 的 TurboMind 推理引擎,运行经由 vllm-project/llm-compressor工具量化后的模型。
目前支持的 llm-compressor 量化模型包括:
- int4 量化(例如 AWQ、GPTQ)
上述量化模型通过 TurboMind 引擎可以在以下 NVIDIA GPU 架构上运行:
| Compute Capability | Micro-architecture | GPUs |
|---|---|---|
| 7.0 | Volta | V100 |
| 7.2 | Volta | Jetson Xavier |
| 7.5 | Turing | GeForce RTX 20 series, T4 |
| 8.0 | Ampere | A100, A800, A30 |
| 8.6 | Ampere | GeForce RTX 30 series, A40, A10 |
| 8.7 | Ampere | Jetson Orin |
| 8.9 | Ada Lovelace | GeForce RTX 40 series, L40, L20 |
| 9.0 | Hopper | H20, H200, H100, GH200 |
| 12.0 | Blackwell | GeForce RTX 50 series |
LMDeploy 将持续跟进并扩展对 llm-compressor 项目的支持。
本文的其余部分由以下章节组成:
模型量化
llm-compressor 提供了丰富的模型量化用例,请参考其教程选择 LMDeploy 支持的量化算法,完成模型量化工作。
LMDeploy 也内置了通过 llm-compressor 对 Qwen3-30B-A3B 进行 AWQ 量化的脚本,供大家进行参考:
# 创建 conda 环境
conda create -n lmdeploy python=3.10 -y
conda activate lmdeploy
# 安装 llm-compressor
pip install llmcompressor
# 下载 lmdeploy 源码,运行量化用用例
git clone https://github.com/InternLM/lmdeploy
cd lmdeploy
python examples/lite/qwen3_30b_a3b_awq.py --work-dir ./qwen3_30b_a3b_awq
在接下来的章节中,我们以此量化模型为例,介绍模型部署、评测精度等方法
模型部署
离线推理
量化后的模型,通过以下几行简单的代码,可以实现离线批处理:
from lmdeploy import pipeline, TurbomindEngineConfig
engine_config = TurbomindEngineConfig()
with pipeline("./qwen3_30b_a3b_4bit", backend_config=engine_config) as pipe:
response = pipe(["Hi, pls intro yourself", "Shanghai is"])
print(response)
关于 pipeline 的详细介绍,请参考这里
在线服务
LMDeploy api_server 支持把模型一键封装为服务,对外提供的 RESTful API 兼容 openai 的接口。以下为服务启动的示例:
lmdeploy serve api_server ./qwen3_30b_a3b_4bit --backend turbomind
服务默认端口是23333。在 server 启动后,你可以通过 openai SDK 访问服务。关于服务的命令参数,以及访问服务的方式,可以阅读这份文档
精度评测
我们将 Qwen3-8B (Dense) 与 Qwen3-30B-A3B (MoE) 的 AWQ 对称/非对称量化模型通过 LMDeploy 部署为服务,并使用 opencompass 在多个学术数据集上评测。结果显示:Qwen3-8B 的非对称量化整体优于对称量化,而 Qwen3-30B-A3B 在两种量化方式间差异不显著;Qwen3-8B 在对称/非对称量化下与 BF16 模型的精度差异小于 Qwen3-30B-A3B。与 BF16 相比,量化模型在长输出数据集,比如 aime2025 (平均 17,635 tokens)、LCB (平均 14,157 tokens),精度下降更明显;在中短输出数据集,比如 ifeval (平均 1,885 tokens),mmlu_pro (平均 2,826),精度符合预期。
| dataset | Qwen3-8B | Qwen3-30B-A3B | ||||
|---|---|---|---|---|---|---|
| bf16 | awq sym | awq asym | bf16 | awq sym | awq asym | |
| ifeval | 85.58 | 83.73 | 85.77 | 86.32 | 84.10 | 84.29 |
| hle | 5.05 | 5.05 | 5.24 | 7.00 | 5.47 | 5.65 |
| gpqa | 59.97 | 56.57 | 59.47 | 61.74 | 57.95 | 57.07 |
| aime2025 | 69.48 | 64.38 | 63.96 | 73.44 | 64.79 | 66.67 |
| mmlu_pro | 73.69 | 71.73 | 72.34 | 77.85 | 75.77 | 75.69 |
| LCBCodeGeneration | 50.86 | 44.10 | 46.95 | 56.67 | 50.86 | 49.24 |
复现方式可以参考这份文档