#175 · 主分类: 推理与本地部署

WebGPT

gpt nanogpt transformers webgpu

使用WebGPU在浏览器上运行GPT模型。这是一个不到约1500行纯Javascript实现的GPT推理。

项目最后更新:01/12/24

GitHub Stars

3.8K

Forks数量

223

贡献者数量

8

许可证

Other

收录理由

WebGPT 用不到 1500 行原生 JavaScript 就把一个 transformer 模型塞进了浏览器标签页,推理全程跑在用户自己的 GPU 上,不需要后端参与。作者把提速的关键决策都摊开写了:核函数融合、键值缓存、缓冲区复用,还记录了不同参数规模下的实际耗时,比如 775M 模型在 2020 款 M1 Mac 上约 120ms/token。仓库自带两个可直接运行的模型,一个是玩具级的 GPT-Shakespeare,另一个是 GPT-2 117M,转换脚本也提供了,想换自己的权重可以照着改。对关心端侧推理或要求权重不出设备的人来说,这份代码既是 WebGPU 能力边界的实测,也是评估当前成本的真实样本。整篇 README 读起来更像工程日志而不是产品宣传,这种坦诚对理解 transformer 推理的实况很有帮助。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目