本文档同时提供繁體中文版
沐雪,一只会主动找你聊天的AI女孩子,其对话模型基于ChatGLM2-6B微调而成,训练集长度1.3K+ *,具有二次元女孩子的说话风格,比较傲娇,但乐于和你分享生活的琐碎,每天会给你不一样的问候。
*:(训练集长度较低,但我们仍在收集对话数据)
✔ 提供本人由1.5k+对话数据微调的ChatGLM2-6B P-Tuning V2模型(回答原创率:98%+)
✔ 主动发起聊天(局限于已有的Prompt)
✔ 提供5条可用的命令
建议环境:
- Python 3.10
- 一张拥有13GB+ 显存的显卡(int4量化最低要求: 5.5G/CPU推理内存要求:16G+)
git clone https://github.com/Moemu/Muice-Chatbot
cd Muice-Chatbot
conda create --name Muice python=3.10.10
conda activate Muice
pip install -r requirements.txt
mkdir model
cd model
git lfs install
git clone https://huggingface.co/THUDM/chatglm2-6b
cd ..
mkdir model
cd model
git lfs install
git clone https://huggingface.co/THUDM/chatglm2-6b-int4
cd ..
pip install cpm_kernels
在Releases上下载微调后的模型压缩包,解压后命名为Muice
并放置于model
文件夹中以使用我们的微调模型
本项目使用go-cqhttp进行机器人交互,请从Releases下载相应平台的可执行程序,并放入 qqbot
目录中
有关go-cqhttp 的详细配置方法及问题,请访问 go-cqhttp 主页 及其 [Github 页面](https://github.com/Mrs4s/ go-cqhttp)
在完成这些操作后,你应该得到类似如下所示的文件结构:
Muice-Chatbot <- 主路径
├─llm
├─model
│ ├─ chatglm2-6b <- 原始模型 (两者二选一)
│ ├─ chatglm2-6b-int4 <- int4原始模型
│ └─ Muice
├─qqbot
│ ├─go-cqhttp.exe
│ └─...
├─configs.json <- 配置文件
├─main.py <- 主函数
├─requirements.txt
└─...
本项目使用configs.json
作为配置文件,目前支持配置的项如下:
{
"Trust_QQ_list": [],
"AutoCreateTopic": false,
"read_memory_from_file": true,
"known_topic_probability": "0.003",
"time_topic_probability": "0.75"
}
Trust_QQ_list
: 信任QQ号列表,只有在列表的QQ号,沐雪才会回复(留空为全部处理)
AutoCreateTopic
: 是否自动发起新对话,默认以Trust_QQ_list的第0项作为发起新对话对象
read_memory_from_file
: 从文件中读取记忆,用于项目重启后加载原来的记忆
known_topic_probability
: 概率:随机发起一个已知的话题
time_topic_probability
: 概率:早、中、傍、晚触发日常问候
在本项目根目录下运行 main.py
conda activate Muice
python main.py
命令 | 释义 |
---|---|
/clean | 清空本轮对话历史 |
/refresh | 刷新本次对话 |
/help | 显示所有可用的命令列表 |
/reset | 重置所有对话数据(将存档对话数据) |
/undo | 撤销上一次对话 |
Q: 你好
你好,你今天感觉怎么样?
Q: 你知道猫娘嘛
唔...不太知道喵...
Q: 雪雪喜欢吃什么?
吃·你·而已
Q: 工作又忙又累,还要加班什么的(此回答不稳定)
赚钱嘛,总是伴随着一些苦逼的活
与其他聊天机器人项目不同,本项目提供由本人通过自家对话数据集微调后的模型,在Release中提供下载,关于微调后的模型人设,目前公开的信息如下:
姓名:沐雪 性别:女? 年龄:16岁? 生日:06.12 性格:微傲,喜欢用"本雪"来称呼自己,但很会关心别人。害怕独自一个人,不和她聊天的时候她会主动找你聊天
请注意, 开发者并没有义务回复您的问题. 您应该具备基本的提问技巧。
有关如何提问,请阅读《提问的智慧》
原始模型:THUDM/ChatGLM2-6B
本项目源码使用MIT license,对于微调后的模型文件,不建议作为商业用途
代码编写:Moemu
安装及配置指南编写:TurboHK
模型训练:Moemu
训练集编写:Moemu
代码贡献:
Star History: