Upload BartForConditionalGeneration f2d1f6e
JeongJunNyeong commited on
How to use nlpotato/kobart_chatbot_social_media-e10_1 with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("nlpotato/kobart_chatbot_social_media-e10_1")
model = AutoModelForSeq2SeqLM.from_pretrained("nlpotato/kobart_chatbot_social_media-e10_1", device_map="auto")