开源了一个 1600 行的大模型推理引擎,能跑真实模型,测试场景吞吐超过 vLLM

21 小时 43 分钟前
 mumu157

想入门大模型推理,直接读 vLLM 太重。所以开源了 LiteMoE:约 1600 行 Python ,能跑真实的 Granite 模型,一个下午能读完主线。

适合想快速上手推理的同学:

固定批次的测试场景下,吞吐是 vLLM 的 105 ~ 109%,脚本可复现。算子用的是 vLLM 的,提升来自算子外的执行层。定位是学习和实验,上线还是用 vLLM 。

如果大家感兴趣,后面会继续做多卡、CPU 版本系列。

项目: https://github.com/ZhuLinsen/LiteMoE 演示: https://zhulinsen.github.io/LiteMoE/visualizer/?lang=zh

444 次点击
所在节点    开源软件
0 条回复

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/1253957

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX