gpt2_chatbot

☕ GPT2 チャットボット

Daily Dialogue 、 Empathetic Dialogues 、 PERSONA-CHAT 、 Blended Skill Talkデータセットでトレーニングされた日常会話用の GPT-2 チャットボット。このチャットボットは、言語モデリングのヘッドを備えた 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モードの 2 つの方法があります。

インタラクションモード

チャットボットを起動するには、次のコマンドを実行します。モデルへの--checkpointパスを指定します

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

トレインモード

モデルをトレーニングするには、次のコマンドを実行します。必要に応じて--checkpoint指定します

python chatbot.py --mode train

ライセンス

著作権 © 2022 xcapt0.
このプロジェクトは MIT ライセンスを受けています。