NM

nvidia/model-optimizer

Developer tools

이 페이지는 AI로 자동 번역되었습니다. 기술적인 세부 사항은 원문을 확인하세요. 원문 보기

5.5천 stars 품질 76 트렌드 57Apache-2.0

양자화, 증류, 가지치기, 신경 구조 탐색, 추측적 디코딩 등 SOTA 모델 최적화 기법의 통합 라이브러리입니다.

개요

문서 | 로드맵 | 발표 블로그 (또는 , 라고 불림)는 양자화, 가지치기, 신경 아키텍처 검색(NAS), 증류, 추측적 디코딩, 희소성 등 최첨단 모델 최적화 기법을 포함한 라이브러리로, 모델을 가속화합니다. 모델 옵티마이저는 현재 Hugging Face, PyTorch 또는 ONNX 모델의 입력을 지원합니다. 모델 옵티마이저는 사용자가 위의 모델 최적화 기법을 쉽게 작성하고 최적화된 양자 체크포인트를 내보낼 수 있도록 파이썬 API를 제공합니다. 모델 옵티마이저는 또한 NVIDIA Megatron-Bridge, Megatron-LM, Hugging Face Accelerate와 통합되어 필요한 추론 최적화 기법을 학습할 수 있습니다. NVIDIA AI 소프트웨어 생태계에 원활하게 통합된 Model Optimizer에서 생성된 양자 체크포인트는 SGLang, TensorRT-LLM, TensorRT, vLLM과 같은 추론 프레임워크에 배포할 준비가 되어 있습니다. 통합된 Hugging Face 내보내기 API는 이제 트랜스포머와 디퓨저 모델 모두를 지원합니다.

개요(AI 번역)

문서 | 로드맵 | 공지 블로그


NVIDIA 모델 옵티마이저(Model Optimizer 또는 ModelOpt로 불림)는 양자화, 가지치기, 신경 아키텍처 검색(NAS), 증류, 가변 디코딩, 희소성 등 최첨단 모델 최적화 기법을 포함한 라이브러리로, 모델을 가속화합니다.

[입력] 모델 옵티마이저는 현재 Hugging Face, PyTorch 또는 ONNX 모델의 입력을 지원합니다.

[최적화] 모델 옵티마이저는 사용자가 위의 모델 최적화 기법을 쉽게 조합하고 최적화된 양자화된 체크포인트를 내보낼 수 있도록 파이썬 API를 제공합니다. 모델 옵티마이저는 또한 NVIDIA Megatron-Bridge, Megatron-LM, Hugging Face Accelerate와 통합되어 필요한 추론 최적화 기법을 학습할 수 있습니다.

[배포용 내보내기] NVIDIA AI 소프트웨어 생태계에 원활하게 통합된 Model Optimizer에서 생성된 양자화된 체크포인트는 SGLang, TensorRT-LLM, TensorRT, vLLM과 같은 추론 추론 프레임워크에 배포할 준비가 되어 있습니다. 통합된 Hugging Face 내보내기 API는 이제 트랜스포머와 디퓨저 모델 모두를 지원합니다.

최신 뉴스

이 페이지는 AI로 자동 번역되었습니다. 기술적인 세부 사항은 원문을 확인하세요.

README

원문 README의 일부입니다. 원문 보기.


NVIDIA Model Optimizer (referred to as Model Optimizer, or ModelOpt) is a library comprising state-of-the-art model optimization techniques including quantization, pruning, Neural Architecture Search (NAS), distillation, speculative decoding and sparsity to accelerate models.

[Input] Model Optimizer currently supports inputs of a Hugging Face, PyTorch or ONNX model.

[Optimize] Model Optimizer provides Python APIs for users to easily compose the above model optimization techniques and export an optimized quantized checkpoint. Model Optimizer is also integrated with NVIDIA Megatron-Bridge, Megatron-LM and Hugging Face Accelerate for training required inference optimization techniques.

[Export for deployment] Seamlessly integrated within the NVIDIA AI software ecosystem, the quantized checkpoint generated from Model Optimizer is ready for deployment in downstream inference frameworks like SGLang, TensorRT-LLM, TensorRT, or vLLM. The unified Hugging Face export API now supports both transformers and diffusers models.

Latest News

GitHub에서 전체 README 보기

추천 도구

다른 키워드를 입력하거나 필터를 제거해 보세요.

설치

npx skillfish add nvidia/model-optimizer
MCP 초보자라면? mcpskill.org 가이드부터 시작하세요 →