gpt2_chatbot

☕ Chatbot GPT2

Chatbot GPT-2 pour les conversations quotidiennes formées sur les ensembles de données Daily Dialogue , Empathetic Dialogues , PERSONA-CHAT et Blended Skill Talk . Ce chatbot est basé sur le transformateur de modèle GPT2 avec une tête de modélisation de langage sur le dessus.

⌛Installation

Téléchargez le modèle depuis le stockage AWS S3 et exécutez la commande suivante :

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

? Usage

Exécutez le conteneur Docker :

docker run --rm -it gpt2_bot

Il existe 2 manières différentes d'utiliser le chatbot : le mode train et le mode interact

Mode d'interaction

Pour lancer le chatbot, exécutez la commande suivante. Spécifiez le chemin --checkpoint vers votre modèle

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

Mode train

Pour entraîner le modèle, exécutez la commande suivante. Spécifiez --checkpoint si vous en avez besoin

python chatbot.py --mode train

Licence

Droits d'auteur © 2022 xcapt0.
Ce projet est sous licence MIT.