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 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.
Этот проект имеет лицензию MIT.