AbaAba-AI
AbaAba-AI copied to clipboard
Local lightweight chat AI based on RWKV(基于RWKV的本地轻量级聊天AI)
Meow-AI
“Meow-AI”是基于RWKV的本地轻量级聊天AI
环境配置
推荐Python版本 3.9.18
需要模块numpy、tokenizers、prompt_toolkit、flask、flask-socketio、torch、subprocess
pip3 install numpy tokenizers prompt_toolkit flask flask-socketio torch subprocess
运行要求
最低4G运行内存(CPU+4G内存,可以运行430m的小模型)
最高不限(最高可运行14b的大模型,聊天质量和效果更加好)
运行设置
根据自己电脑配置在模型网站(https://huggingface.co/BlinkDL)下载合适的模型 (镜像网址)
下载的所有文件放在models文件夹内,并修改config.json中的“modelFile”,将'RWKV-x060-World-1B6-v2-20240208-ctx4096'修改为自己下载的模型名
运行main.py,在浏览器中打开http://172.0.0.1:5000 即可开始对话
功能更新
- 2024-01-05:支持手动停止对话
- 2024-01-19:支持编辑对话实现简单的手动微调
- 2024-01-23:支持配置导入/导出,支持自定义AI性格人设
- 2024-03-01:完全使用RWKV架构,实现更小的模型体积,降低运行内存和CPU占用
- 2024-04-26:优化整体架构
- 2024-07-08:支持扩展功能,支持微信自动聊天
演示效果
1. 持续对话
该演示环境为GPU+8G,使用'RWKV-5-World-3B-v2-20231118-ctx16k'模型
2. 手动调整对话
~注:手动添加修改更多的自己预设的对话,对后续的实现自己想要的聊天效果有很大帮助~
3. 配置导入导出
可以导入导出对话以及配置信息
4. 调教AI人设性格
可以自定义设置MeowAI人设性格,让MeowAI更加符合自己的喜好
4. 扩展功能
支持添加自定义扩展功能,目前已完成“微信自动聊天”
点击启动后,将自动打开微信,自动回复聊天
未来展望
- 完成更多功能扩展
- 实现本地一键 懒人包,不用配置环境一键启动即可运行
- 待续……