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은 다양한 양자화 기술(GPTQ, AWQ, ParoQuant, GGUF, FP8, EXL3, QQQ 등)을 사용하여 대규모 언어 모델(LLM)을 압축하여 더 작고 빠른 버전으로 변환하는 Python 라이브러리입니다. 단일하고 일관된 API를 제공합니다:
- 캘리브레이션 – 샘플 데이터셋에서 모델의 활성화 통계를 측정합니다.
- 양자화 – 가중치를 저비트 형식(2‑8비트, FP8 등)으로 변환합니다.
- 검증 – 양자화 후에도 모델의 품질이 유지되는지 확인합니다.
- 변환 및 저장 – GGUF 또는 MLX와 같은 형식으로 체크포인트를 내보냅니다.
- 가속 추론 – vLLM 및 SGLang과 같은 인기 있는 런타임으로 양자화된 모델을 로드합니다.
이 모든 것은 NVIDIA CUDA, AMD ROCm, Huawei Ascend, Intel XPU, Apple 실리콘 Metal, 일반 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]' # Apple 실리콘용 선택적 MLX
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, Alibaba Cloud의 공개 참조가 나열되어 있으며, 이는 라이브러리가 프로덕션 등급 파이프라인 및 모델 호스팅 서비스에서 사용되고 있음을 나타냅니다.
위의 모든 정보는 저장소의 README에서 직접 가져온 것이며, 추가 기능은 추론되지 않았습니다.
관련
- 프로젝트
- 프로젝트
- Dispatch
- 프로젝트
- 프로젝트