察元AI教程库 · 文章 · 部署
大语言 语言模型(LLM)推理包含两个阶段:提示(或预填充)阶段以生成首个标记,以及扩展(或解码)阶段以生成后续标记。在本研究中,我们提出了一种高效的并行化方案KV-Runahead,以加速预填充阶段。关键观察是
作者:Minsik Cho, Mohammad Rastegari, Devang Naik · 语言:en
大语言 语言模型(LLM)推理包含两个阶段:提示(或预填充)阶段以生成首个标记,以及扩展(或解码)阶段以生成后续标记。在本研究中,我们提出了一种高效的并行化方案KV-Runahead,以加速预填充阶段。关键观察是