gpt2_chatbot

☕ Bot Obrolan GPT2

Chatbot GPT-2 untuk percakapan sehari-hari dilatih pada Daily Dialogue , Empathetic Dialogues , PERSONA-CHAT , kumpulan data Blended Skill Talk . Chatbot ini dibuat berdasarkan trafo Model GPT2 dengan kepala pemodelan bahasa di atasnya.

⌛ Instalasi

Unduh model dari penyimpanan AWS S3 dan jalankan perintah berikut:

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

? Penggunaan

Jalankan wadah buruh pelabuhan:

docker run --rm -it gpt2_bot

Ada 2 cara berbeda untuk menggunakan chatbot: mode train dan interact

Modus interaksi

Untuk meluncurkan chatbot jalankan perintah berikut. Tentukan jalur --checkpoint ke model Anda

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

Modus kereta

Untuk melatih model, jalankan perintah berikut. Tentukan --checkpoint jika Anda perlu

python chatbot.py --mode train

Lisensi

Hak Cipta © 2022 xcapt0.
Proyek ini berlisensi MIT.