gpt2_chatbot

☕ GPT2 แชทบอท

GPT-2 chatbot สำหรับการสนทนารายวันที่ได้รับการฝึกอบรมเกี่ยวกับ 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

การใช้แชทบอทมี 2 วิธี: โหมด train และ interact

โหมดการโต้ตอบ

หากต้องการเปิด Chatbot ให้รันคำสั่งต่อไปนี้ ระบุ --checkpoint path ไปยังโมเดลของคุณ

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

โหมดรถไฟ

เพื่อฝึกโมเดลให้รันคำสั่งต่อไปนี้ ระบุ --checkpoint หากคุณต้องการ

python chatbot.py --mode train

ใบอนุญาต

ลิขสิทธิ์ © 2022 xcapt0.
โครงการนี้ได้รับอนุญาตจาก MIT