单卡 RTX 5090 四夜训练 EAGLE-3 投机解码头,Gemma-3-27B 推理提速超 50%

max_paperclips · x · 2026-08-11

开发者 witcheer 分享了如何使用单张 RTX 5090 显卡,在四个晚上从零训练出一个针对 Gemma-3-27B 的 EAGLE-3 投机解码头(drafter)。

该解码头基于 716M 参数的单层 Llama 架构,结合 sglang 框架进行验证。实测显示,在完全不改变输出质量的前提下,模型推理获得了显著加速:

作者还对比了不同的树状推测配置,指出 tree-3-4-8 提供了最佳的净加速效果,并已将模型在 Hugging Face 上开源。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →