gpustack/gguf-parser-go

Review/Check GGUF files and estimate the memory usage and maximum tokens per second.

解決する課題

GGUF Parserを使用すると、GGUFモデルファイルを検査し、ファイル全体をダウンロードしたりモデルを実行したりすることなく、ハードウェア要件を推定できます。モデルがシステムのメモリに収まるかどうか、またどのようなパフォーマンスが期待できるかを迅速に判断するのに役立ちます。

仕組み

このツールはチャンク読み取りを使用して、ローカルファイルまたはリモートURL(HuggingFace、ModelScope、Ollamaを含む)からメタデータを直接解析します。ファイルの必要な部分のみを読み取ることで、完全なダウンロードを回避します。次に、メモリ使用量(RAM/VRAM)を計算し、特定のハードウェア構成に基づいて、1秒あたりのトークン数などのパフォーマンス指標を推定します。

対象ユーザー

このツールは、GGUF形式のモデルを扱い、デプロイ前にモデルの互換性とリソース割り当てを高速かつ効率的に検証する必要がある開発者やAIエンジニア向けに設計されています。

ハイライト

  • 完全なダウンロード不要: チャンク読み取りを使用して、モデル全体をダウンロードせずにリモートファイルを検査します。
  • 高精度なメモリ推定: メモリ使用量を高精度で予測し、通常わずか約100MiBの誤差です。
  • パフォーマンス予測: 提供されたデバイス指標に基づいて、1秒あたりの最大トークン数を推定します。
  • 多様な互換性: 埋め込み、再ランキング、決定モデル、およびMLAやSWAなどの特殊なアーキテクチャを含むさまざまなモデルタイプをサポートします。
  • 効率的なパフォーマンス: 高速で軽量な実行のためにGoで記述されています。

関連

  • プロジェクト
  • Dispatch
  • Dispatch
  • プロジェクト
  • プロジェクト