ModelCloud/GPTQModel
LLM model quantization (compression) toolkit with HW acceleration support for Nvidia, AMD, Intel GPU and Intel/AMD/Apple CPU via HF, vLLM, and SGLang.
GPT‑QModel – LLM量化统一平台
功能
GPT‑QModel是一个Python库,允许您使用各种量化技术(GPTQ、AWQ、ParoQuant、GGUF、FP8、EXL3、QQQ等)压缩大型语言模型(LLM),使其更小、运行更快。它提供统一一致的API,用于:
- 校准 – 在样本数据集上测量模型的激活统计。
- 量化 – 将权重转换为低比特格式(2‑8位、FP8等)。
- 验证 – 检查量化后的模型是否保持质量。
- 转换与保存 – 以GGUF或MLX等格式导出检查点。
- 加速推理 – 使用vLLM和SGLang等流行运行时加载量化模型。
所有这些都可以在多种硬件后端上运行:NVIDIA CUDA、AMD ROCm、华为Ascend、Intel XPU、Apple硅金属和普通CPU。
主要功能(根据README描述)
| 功能 | 详情 |
|---|---|
| 广泛的方法支持 | GPTQ、AWQ、ParoQuant、QQQ、GGUF、FP8、EXL3、GPTAQ、FOEM、BitsAndBytes等。 |
| 多运行时 | 与🤗 Transformers、Optimum、PEFT原生集成,推理后端vLLM和SGLang。 |
| 硬件覆盖 | CUDA、ROCm、Ascend NPU、Intel XPU、Apple M系列(Metal/MLX)和CPU(AVX/AMX)。 |
| 动态逐层量化 | 每层可以有自己的位宽或完全排除。 |
| 多GPU数据并行量化 | 在GPU之间分散工作时速度提升最高约80 %。 |
| Apple硅原生流水线 | 在M系列芯片上进行基于MLX的GPTQ/AWQ量化和EXL3权重量化。 |
| 广泛的模型目录 | 通过内部MODEL_MAP支持200多个模型系列(Llama、Gemma、Qwen、DeepSeek、InternVL等)。 |
| 打包API | 用于逐层内核可行性的hf_打包/重新打包API(被AutoRound等工具使用)。 |
| 高性能内核 | Marlin、Machete、BitBLAS、Triton、Exllama V3/EXL3和Intel Torch融合内核。 |
| 注重质量 | 单元测试覆盖、检查点验证和4位GPTQ评估工具。 |
| 开源和PyPI | Apache‑2.0许可证,在GitHub和PyPI上发布(pip install gptqmodel)。 |
谁可能使用它?
- 研究工程师 – 希望试验低比特LLM而无需拼凑许多单独工具的人。
- MLOps团队 – 需要在异构集群上量化、验证和部署模型的可重现流水线的团队。
- 产品开发者 – 面向内存有限的边缘设备(Apple Silicon、Intel CPUs、Ascend NPU)。
- 社区贡献者 – 构建新量化方法的人 – 模块化的“方法 → 格式 → 后端”矩阵使自定义内核易于集成。
如何开始(README中的快速示例)
pip install 'gptqmodel[mlx]' # 可选的MLX用于Apple硅
from gptqmodel import GPTQModel
# 在Apple硅上使用原生MLX流水线量化Qwen模型
GPTQModel.quantize_mlx(
"Qwen/Qwen3-0.6B",
"./qwen3-mlx-gptq",
method="gptq", # 或"awq"
)
相同的API适用于CUDA、ROCm或CPU后端;只需安装相应的附加包并选择所需的method和format。
生态系统与采用
README列出了Hugging Face、Intel、NVIDIA和阿里云的公开参考,表明该库正被用于生产级流水线和模型托管服务。
上述所有信息均直接来自仓库的README;没有推断其他功能。
相关
- 项目
- 项目
- Dispatch
- 项目
- 项目