gpt2_chatbot

☕ GPT2 聊天机器人

用于日常对话的 GPT-2 聊天机器人，在Daily Dialogue 、 Empathetic Dialogues 、 PERSONA-CHAT 、 Blended Skill Talk数据集上进行训练。该聊天机器人是基于 GPT2 模型转换器制作的，顶部有语言建模头。

⌛ 安装

从 AWS S3 存储下载模型并运行以下命令：

git pull https://github.com/xcapt0/gpt2_chatbot.git
docker build -t gpt2_bot . 

？用法

运行docker容器：

docker run --rm -it gpt2_bot

有两种不同的方式使用聊天机器人： train和interact模式

互动方式

要启动聊天机器人，请运行以下命令。指定模型的--checkpoint路径

python chatbot.py --mode interact --checkpoint path/to/model.h5

火车模式

要训​​练模型，请运行以下命令。如果需要，请指定--checkpoint

python chatbot.py --mode train

执照

版权所有 © 2022 xcapt0。
该项目已获得麻省理工学院许可。