之前总是觉得,单片机上最多也就跑个几十万参数的小玩具模型。

乌克兰有个牛人,直接把 2890万参数 的模型塞进了 ESP32-S3

为了更适配, 模型是作者自己训练的。

而且,跑的速度还很快。

打开网易新闻 查看精彩图片

可玩性还是很强的,现在只是接了个小屏幕,后面可能会有很多拓展。

esp32-ai是一个让2890万参数模型完全运行在ESP32-S3上的开源项目。

模型完全离线推理,并且能以 9.5 tokens/秒速度在芯片自带的小屏幕上实时输出文字。 项目借用了Google Gemma模型中的层间嵌入技术, 把大部分参数存放在Flash中,只在需要时按需读取,突破了单片机极有限的SRAM限制。

模型基于TinyStories数据集训练,能生成简短且基本连贯的故事。

这生成速度真的很快了。

AI类的硬件,比如一些AI对话玩具,都还是得联网,虽然联网有联网的好处,但是不联网也有不联网的需求。

今天这个项目也是个方向,肯定会有便宜又好玩的AI硬件完全运行在本地。

之前大家认知里就觉得单片机推不动大一点的模型,所以这个项目我觉得更大的价值是证明了单片机推的模型也能很流畅的讲出一个儿童故事。

起码AI儿童故事的AI硬件设备是可以玩玩了。

还有可能有更多简单的AI需求,比如做一些简单的数据比对等等,都可以实现了。