ReplaySSM 优化 Mamba2 推理:vLLM 支持 SSM 输入缓存与 MTP
vLLM 和 NVIDIA 把 Mamba2 的解码内存开销压下来了,用 SSM 状态模型跑长文本的可以升级试试。
vLLM 和 NVIDIA 把 Mamba2 的解码内存开销压下来了,用 SSM 状态模型跑长文本的可以升级试试。
这篇论文解决了长上下文下投机解码的瓶颈,用滑动窗口草稿省掉99%的KV读取,实测速度提升近三成,而且不损失精度。搞大模型推理优化的值得看。
做视觉语言模型部署或实时推理的开发者,Zamba2-VL 的首 token 延迟优势能显著提升用户体验,值得直接尝试。