
乌克兰有个牛人,直接把2890万参数的模型塞进了ESP32-S3。
为了更适配,模型是作者自己训练的。
而且,跑的速度还很快。

可玩性还是很强的,现在只是接了个小屏幕,后面可能会有很多拓展。
项目简介
模型完全离线推理,并且能以9.5 tokens/秒的速度在芯片自带的小屏幕上实时输出文字。项目借用了Google Gemma模型中的层间嵌入技术,把大部分参数存放在Flash中,只在需要时按需读取,突破了单片机极有限的SRAM限制。
模型基于TinyStories数据集训练,能生成简短且基本连贯的故事。
DEMO
一些想法
今天这个项目也是个方向,肯定会有便宜又好玩的AI硬件完全运行在本地。
之前大家认知里就觉得单片机推不动大一点的模型,所以这个项目我觉得更大的价值是证明了单片机推的模型也能很流畅的讲出一个儿童故事。
起码AI儿童故事的AI硬件设备是可以玩玩了。
还有可能有更多简单的AI需求,比如做一些简单的数据比对等等,都可以实现了。
项目链接
https://github.com/slvDev/esp32-ai
© 版权声明
文章版权归作者所有,未经允许请勿转载。
