AI在线 AI在线

SwiftInfer

多轮对话推理速度提升46%，开源方案打破LLM多轮对话的长度限制​

多轮对话推理速度提升46%，开源方案打破LLM多轮对话的长度限制​
在大型语言模型（LLM）的世界中，处理多轮对话一直是一个挑战。前不久麻省理工 Guangxuan Xiao 等人推出的 StreamingLLM，能够在不牺牲推理速度和生成效果的前提下，可实现多轮对话总共 400 万个 token 的流式输入，22.2 倍的推理速度提升。但 StreamingLLM 使用原生 PyTorch 实现，对于多轮对话推理场景落地应用的低成本、低延迟、高吞吐等需求仍有优化空间。Colossal-AI 团队开源了 SwiftInfer，基于 TensorRT 实现了 StreamingLLM
1/8/2024 3:22:00 PM
机器之心
  • 1