JetBrains 开源 Mellum2.1 Thinking MTP 头,解码提速 1.4-1.8 倍

pmttyji · reddit · 2026-11-02

JetBrains 在 Hugging Face 发布 Mellum2.1-12B-A2.5B-Thinking 的 MTP(multi-token prediction)版本:额外一个 transformer 块与主模型联合训练,推理时作为 speculative draft,每 token 均被主模型验证,输出不变的前提下解码提速 1.4–1.8 倍,草稿接受率约 90%。

权重存于 model-mtp.safetensors(mtp. 前缀),config.json 中以 numnextnpredictlayers: 1 声明。各推理栈用法:

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →