github.devotd.
新发现推理与服务

nvidia/model-optimizer 0.47.0 — 用于推理的模型优化库

nvidia/model-optimizer 是一个 Python 库,通过多种模型优化技术生成可部署到推理框架的检查点。

仓库信息nvidia/model-optimizer ↗

nvidia/model-optimizer 是一个 Python 库,通过多种模型优化技术生成可部署到推理框架的检查点。

GitHub 项目说明: A unified library of SOTA model optimization techniques like quantization, distillation, pruning, neural architecture search, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.

星标
★ 5.4K
复刻
757
语言
Python
许可证
Apache-2.0
最新版本
0.47.0
查看原始来源 ↗在 GitHub 查看 ↗

nvidia/model-optimizer 是一个帮助压缩深度学习模型以用于推理部署的 Python 库。最新版本为 0.47.0,仓库说明称其提供量化、剪枝、蒸馏、神经架构搜索(NAS)、投机解码等技术。本次发布中具体有哪些新变化,所提供的资料中没有说明。该仓库以 Apache-2.0 许可证公开。

该库接受 Hugging Face、PyTorch 或 ONNX 模型作为输入,在 Python API 中组合优化技术,并可导出优化后的量化检查点。量化是用较低精度表示权重的方式,资料中给出的目的是减少检查点在推理引擎中所占用的权重内存。导出的检查点可部署到 SGLang、TensorRT-LLM、TensorRT、vLLM 等推理框架。因此它关注的是把模型压缩并适配到推理环境进行部署的过程,而不是训练本身。

不过,所提供的资料中没有特定模型上的压缩率、准确率变化、吞吐量等实测结果。技术和支持对象可以确认,但实际性能变化的幅度仅凭这些信息无法判断。最新版本号可以确认,但该版本的变更内容也未给出,因此无法断定新增了哪些内容。