Открытая Python-библиотека Intel для сжатия моделей глубокого обучения с помощью квантования, разреживания и дистилляции в PyTorch, JAX и ONNX.
NVIDIA/Model-Optimizer
Библиотека NVIDIA с открытым исходным кодом для сжатия моделей глубокого обучения с помощью квантования, pruning, дистилляции, разреженности и спекулятивного декодирования, ускоряющих инференс.
- Звёзды
- 5,553
- Форки
- 803
- Открытые issues
- 524
- Последний коммит
- 3 часа назад
Что это?
NVIDIA Model Optimizer (ModelOpt) — это библиотека Python, которая применяет методы оптимизации моделей: посттренировочное квантование, обучение с учетом квантования, pruning, дистилляцию знаний, Neural Architecture Search, разреженность и спекулятивное декодирование. На вход она принимает модели Hugging Face, PyTorch или ONNX и предоставляет API Python для комбинирования этих методов и экспорта оптимизированного квантованного чекпойнта.
Экспортированные чекпойнты предназначены для последующего инференса с помощью таких фреймворков, как TensorRT-LLM, TensorRT, vLLM и SGLang. Для оптимизации во время обучения проект также интегрируется с Megatron-LM, Megatron-Bridge и Hugging Face Accelerate, а API экспорта в Hugging Face поддерживает модели transformers и diffusers.
В репозитории есть примеры для LLM, VLM, диффузионных моделей и ONNX, а также матрица поддержки для разных типов моделей и набор предварительно квантованных чекпойнтов, опубликованных на Hugging Face.
Ключевые возможности
- 01Посттренировочное квантование и обучение с учетом квантования, в том числе в форматах NVFP4 и FP8
- 02Структурированный pruning и дистилляция знаний для уменьшения размера LLM
- 03Спекулятивное декодирование и методы разреженности
- 04Экспорт квантованных чекпойнтов для TensorRT-LLM, vLLM и SGLang
- 05Поддержка моделей Hugging Face, PyTorch и ONNX на входе
Сценарии использования
- Квантование крупной LLM с Hugging Face в NVFP4 или FP8 перед запуском с помощью vLLM или TensorRT-LLM
- Pruning и дистилляция крупной модели для создания меньшей модели, сохраняющей большую часть качества
- Добавление чернового модуля для ускорения генерации токенов с помощью спекулятивного декодирования
- Снижение задержки диффузионных моделей в конвейерах генерации изображений или видео
Быстрый старт
- 1
Установите стабильную версию из PyPI
pip install -U nvidia-modelopt[all]
Что важно знать
- При установке также устанавливаются дополнительные сторонние проекты с открытым исходным кодом. Ознакомьтесь с условиями их лицензий.
- Образы контейнеров NVIDIA, с которыми можно использовать это ПО, имеют собственные условия лицензии. Ознакомьтесь с ними перед использованием.
Темы
Частые вопросы
- Что такое Model-Optimizer?
- Библиотека NVIDIA с открытым исходным кодом для сжатия моделей глубокого обучения с помощью квантования, pruning, дистилляции, разреженности и спекулятивного декодирования, ускоряющих инференс. NVIDIA Model Optimizer (ModelOpt) — это библиотека Python, которая применяет методы оптимизации моделей: посттренировочное квантование, обучение с учетом квантования, pruning, дистилляцию знаний, Neural Architecture Search, разреженность и спекулятивное декодирование. На вход она принимает модели Hugging Face, PyTorch или ONNX и предоставляет API Python для комбинирования этих методов и экспорта оптимизированного квантованного чекпойнта.
- Model-Optimizer можно использовать бесплатно? Какая у него лицензия?
- Model-Optimizer распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- Model-Optimizer всё ещё поддерживается?
- Активно поддерживается. Последний коммит в Model-Optimizer — 7 нояб. 2026 г..
- Для чего можно использовать Model-Optimizer?
- Квантование крупной LLM с Hugging Face в NVFP4 или FP8 перед запуском с помощью vLLM или TensorRT-LLM Pruning и дистилляция крупной модели для создания меньшей модели, сохраняющей большую часть качества Добавление чернового модуля для ускорения генерации токенов с помощью спекулятивного декодирования Снижение задержки диффузионных моделей в конвейерах генерации изображений или видео
- Как установить Model-Optimizer?
- Установите стабильную версию из PyPI: pip install -U nvidia-modelopt[all]
- Где скачать Model-Optimizer?
- Исходный код и инструкции по установке — на GitHub: https://github.com/NVIDIA/Model-Optimizer
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
SliceGPT сжимает трансформерные модели, включая LLM, после обучения, удаляя строки и столбцы матриц весов.
Бэкенд Triton Inference Server для запуска TensorRT-LLM и совместимых с HuggingFace LLM через HTTP- и gRPC API Triton.
Python-библиотека для сжатия LLM с помощью квантования и прунинга, создающая чекпойнты, которые vLLM загружает напрямую для инференса.
Единая исследовательская среда для сравнения сжатия, извлечения, объединения и квантования KV-кэша при инференсе LLM с длинным контекстом.
Инструментарий для квантования больших языковых моделей (GPTQ, AWQ, GGUF, FP8, EXL3 и другие) и запуска на оборудовании NVIDIA, AMD, Intel, Apple и CPU.