本项目受到早苗bot启发,且无法忍受青客云屑AI,因此在GitHub上寻找代替方案
默认环境python3.8.10
使用前请先安装运行库:
pip install -r requirements.txt -i https://pypi.douban.com/simple
本项目由yangjianxin1大佬的GPT2-chitchat项目修改(如有侵权联系删除)
仅在原项目上增加使用Flask库进行网页交互,几乎照抄
仅使用cpu进行对话生成,由于闲聊对话生成的内容长度不是很长,因此生成部分在CPU上跑速度也挺快的
模型 | 共享地址 | 模型描述 |
---|---|---|
model_epoch40_50w | 百度网盘【提取码:ju6m】 或 GoogleDrive | 使用50w多轮对话语料训练了40个epoch,loss降到2.0左右。 |
将下载好的模型放入model/model_epoch40_50w文件夹中即可
如需训练自己的GPT-2模型详见原项目
例如:
http://127.0.0.1:5000/?key_word=我喜欢你
答:我也喜欢你