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),并根据您的特定硬件配置估算性能指标,如每秒 token 数。

适用人群

该工具专为处理 GGUF 格式模型的开发者和 AI 工程师设计,需要在部署前快速高效地验证模型兼容性和资源分配。

亮点

  • 无需完整下载:使用分块读取检查远程文件,无需下载整个模型。
  • 准确的内存估算:高精度预测内存使用量,通常偏差仅为约 100MiB。
  • 性能预测:根据提供的设备指标估算每秒最大 token 数。
  • 广泛兼容性:支持多种模型类型,包括嵌入、重排序和决策模型,以及 MLA 和 SWA 等专用架构。
  • 高效性能:采用 Go 编写,执行快速且轻量。

相关

  • 项目
  • Dispatch
  • Dispatch
  • 项目
  • 项目