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。
該計畫已獲得麻省理工學院許可。