gpt2_chatbot

☕ Chatbot GPT2

Chatbot GPT-2 para conversaciones diarias entrenado en conjuntos de datos de Daily Dialogue , Empathetic Dialogues , PERSONA-CHAT y Blended Skill Talk . Este chatbot está hecho en base al transformador modelo GPT2 con un cabezal de modelado de lenguaje en la parte superior.

⌛ Instalación

Descargue el modelo del almacenamiento AWS S3 y ejecute el siguiente comando:

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

? Uso

Ejecute el contenedor acoplable:

docker run --rm -it gpt2_bot

Hay 2 formas diferentes de utilizar el chatbot: train y modo interact

Modo de interacción

Para iniciar el chatbot, ejecute el siguiente comando. Especifique la ruta --checkpoint a su modelo

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

Modo tren

Para entrenar el modelo, ejecute el siguiente comando. Especifique --checkpoint si lo necesita

python chatbot.py --mode train

Licencia

Copyright © 2022 xcapt0.
Este proyecto tiene licencia del MIT.