英特尔开源的 Python 模型压缩库,通过量化、稀疏化和蒸馏压缩深度学习模型。
vllm-project/llm-compressor
一个用于压缩大语言模型的 Python 库,通过量化和剪枝生成可直接在 vLLM 中推理的模型。
- Star
- 3,903
- Fork
- 707
- 未关闭 Issue
- 124
- 最近提交
- 3小时前
这是什么?
LLM Compressor 是一个在使用 vLLM 部署前优化大语言模型的库。它支持对权重、激活值、KV Cache 和注意力进行量化,也能对混合专家(MoE)模型做 REAP 专家剪枝。压缩后的模型以 compressed-tensors 格式保存,vLLM 可以直接加载,无需额外转换。
该库与 Hugging Face 的模型和仓库兼容,提供多种算法,包括朴素的 PTQ(Round-to-Nearest)、GPTQ、AWQ、SmoothQuant、AutoRound,以及 SpinQuant、QuIP 等基于旋转的方法。支持的精度涵盖 FP8、INT8、NVFP4/MXFP4 等微缩放格式,以及 4 位和 8 位的仅权重量化。
针对超大模型,它支持 DDP 分布式量化、顺序加载(sequential onloading)和磁盘卸载,使有限硬件也能完成压缩。常见场景只需调用一次 `oneshot` 并传入 recipe 即可完成量化流程。
核心功能
- 01支持权重、激活值、KV Cache 和注意力的量化,格式包括 FP8、INT8、NVFP4 和 MXFP4。
- 02提供 GPTQ、AWQ、SmoothQuant、AutoRound 和 Round-to-Nearest PTQ 等算法。
- 03输出 compressed-tensors 格式的模型,vLLM 可直接加载。
- 04兼容 Hugging Face 模型与仓库。
- 05支持 DDP、顺序加载和磁盘卸载,便于压缩超大模型。
- 06支持面向 MoE 模型的 REAP 专家剪枝。
适用场景
- 压缩模型显存占用,使其能在更少的 GPU 上通过 vLLM 提供服务。
- 为新发布的开源权重模型生成 FP4 或 FP8 量化版本用于生产推理。
- 在硬件有限的情况下,借助磁盘卸载或 DDP 压缩超大模型。
- 通过 recipe 切换不同量化算法和方案进行对比实验。
标签
常见问题
- llm-compressor 是什么?
- 一个用于压缩大语言模型的 Python 库,通过量化和剪枝生成可直接在 vLLM 中推理的模型。 LLM Compressor 是一个在使用 vLLM 部署前优化大语言模型的库。它支持对权重、激活值、KV Cache 和注意力进行量化,也能对混合专家(MoE)模型做 REAP 专家剪枝。压缩后的模型以 compressed-tensors 格式保存,vLLM 可以直接加载,无需额外转换。
- llm-compressor 可以免费使用吗?用的什么许可证?
- llm-compressor 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
- llm-compressor 还在维护吗?
- 活跃维护中。llm-compressor 最近一次提交是在 2026年10月22日。
- llm-compressor 适合用来做什么?
- 压缩模型显存占用,使其能在更少的 GPU 上通过 vLLM 提供服务。 为新发布的开源权重模型生成 FP4 或 FP8 量化版本用于生产推理。 在硬件有限的情况下,借助磁盘卸载或 DDP 压缩超大模型。 通过 recipe 切换不同量化算法和方案进行对比实验。
- 在哪里获取 llm-compressor?
- 源代码和安装说明在 GitHub 上:https://github.com/vllm-project/llm-compressor
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
一个 Python 神经网络压缩框架,通过量化、剪枝和稀疏化加速 OpenVINO 推理,支持 PyTorch 与 ONNX 模型。
面向大模型分布式训练与推理的深度学习优化库,提升效率并降低使用门槛。
Intel 开源的 LLM/VLM 训练后量化工具包,可将模型压缩为低比特权重,支持 Transformers、vLLM 和 SGLang。
XHToken 推出的开源 4B 与 1.7B 通用语言模型,面向端侧部署和智能体工具调用场景。
面向 LLM 推理的 KV 缓存管理层,可跨推理引擎存储、共享和复用缓存,降低首 token 延迟。