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
- 项目
- 项目