drumih/turbo-fieldfare

Gemma 4 26B-A4B inference in ~2 GB of RAM on any M-series MacBook

解决的问题

TurboFieldfare 使得在仅有 8 GB 内存的 Apple Silicon Mac 上运行庞大的 260 亿参数 Gemma 4 模型成为可能。通常情况下,加载此类模型需要更多的内存;本项目通过按需从 SSD 流式传输必要的模型组件,绕过了这一限制。

工作原理

运行时不会将整个模型加载到内存中,而是将一个小核心和 KV cache 保留在 RAM 中。它使用自定义的 Swift 和 Metal 实现,动态获取并缓存每个 token 生成步骤所需的特定“专家”权重。它采用分块预填充(chunked prefill)来优化提示处理,并使用有界缓存策略在从磁盘流式传输数据时管理内存使用。

适用人群

本项目面向对端侧机器学习感兴趣的开发者和研究人员,特别是那些希望在资源受限的 Apple Silicon 硬件上运行大规模模型的用户。它非常适合想要尝试自定义 Metal 内核和内存高效推理技术的用户。

亮点

  • 在约 2 GB 的 RAM 预算内运行 26B Gemma 4 模型。
  • 包含原生 macOS 应用、CLI 以及兼容 OpenAI 的本地服务器。
  • 具有流式安装程序,可直接将模型权重重新打包为自定义格式,无需暂存完整的检查点。
  • 使用自定义 Metal 内核构建,支持量化操作、MoE (Mixture of Experts) 路由和高效注意力机制。
  • 在 M2 或更新的 Apple Silicon 芯片上,通过配套包支持可选的视觉功能。

相关

  • Dispatch
  • Dispatch
  • Dispatch
  • 项目
  • Dispatch