Skip to content

haven-jeon/KoBART-chatbot

Repository files navigation

Simple Chit-Chat based on KoBART

Purpose

Architecture

drawing

Install

# KoBART 설치 
pip install git+https://github.com/SKT-AI/KoBART#egg=kobart
pip install pandas
pip install pytorch_lightning==1.2.1
git clone --recurse-submodules https://github.com/haven-jeon/KoBART-chatbot.git

How to Train

  1. KoBART 모델 다운로드
>>> from kobart import get_pytorch_kobart_model, get_kobart_tokenizer
>>> get_kobart_tokenizer(".")
>>> get_pytorch_kobart_model(cachedir=".")
  1. chit-chat fine-tuning
$ python kobart_chit_chat.py  --gradient_clip_val 1.0 --max_epochs 3 --default_root_dir logs --model_path kobart_from_pretrained  --tokenizer_path emji_tokenizer --chat --gpus 1

Chat!

user > 나 정말 ㅜㅜ.  
Simsimi >  무슨 일 있었나봐요.
user > 나 정말 🤣
Simsimi >  그런 사람는 거르세요.
user > 나 정말 ^^
Simsimi >  칭찬이네요.
user > 너무 어두워.
Simsimi >  빛이 있을 거예요.
user > 어버이날엔 뭘 하죠?
Simsimi >  어버이날은 가족들과 함께 보내는 시간이니 만큼 가족들과 함께 하는 시간이 좋을 거예요.
user > 큰돈을 만드는 방법은?
Simsimi >  돈은 쓴만큼 또 생긴다고 하던데요.
user > 집은 너무 비싼거 같아요.
Simsimi >  돈을 얼른 모아야할 이유가 생겼네요.

Colab tutorials

  • Open In Colab

About

KoBART chatbot

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published