察元AI教程库 · 文章 · 部署
大语言模型(LLMs)的推理过程计算需求很高。为了降低自回归解码的开销,采用键值(KV)缓存来存储中间激活值,显著降低了生成每个标记的计算开销。然而,KV缓存所需的内存却成为一个瓶颈。
作者:Chaoyi Jiang, Lei Gao, Hossein Entezari Zarch · 语言:en
大语言模型(LLMs)的推理过程计算需求很高。为了降低自回归解码的开销,采用键值(KV)缓存来存储中间激活值,显著降低了生成每个标记的计算开销。然而,KV缓存所需的内存却成为一个瓶颈。