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シリコンメタル、通常の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
  • プロジェクト
  • プロジェクト