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 모드의 두 가지가 있습니다.

상호작용 모드

챗봇을 시작하려면 다음 명령을 실행하세요. 모델에 --checkpoint 경로를 지정하세요.

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

열차 모드

모델을 학습하려면 다음 명령어를 실행하세요. 필요한 경우 --checkpoint 지정하십시오.

python chatbot.py --mode train

특허

저작권 © 2022 xcapt0.
이 프로젝트는 MIT 라이선스를 받았습니다.