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

هناك طريقتان مختلفتان لاستخدام chatbot: وضع train interact

وضع التفاعل

لتشغيل chatbot قم بتشغيل الأمر التالي. حدد --checkpoint مسار نقطة التفتيش إلى النموذج الخاص بك

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

وضع القطار

لتدريب النموذج قم بتشغيل الأمر التالي. حدد --checkpoint إذا كنت بحاجة

python chatbot.py --mode train

رخصة

حقوق النشر © 2022 xcapt0.
هذا المشروع مرخص من معهد ماساتشوستس للتكنولوجيا.