에러 도저히 못 고칠거 같아서 그냥 올린다. 아무나 고칠 수 있는 사람은 가져가서 활용해주세요.. import torch from transformers import GPT2LMHeadModel, GPT2Tokenizer #모델 다운로드 model_name = 'gpt2-medium' tokenizer = GPT2Tokenizer.from_pretrained(model_name) model = GPT2LMHeadModel.from_pretrained(model_name) #프롬 prompt = "대화가 시작되었습니다.
나가기는 Ctrl + C" prompt_length = len(tokenizer.encode(prompt, add_special_tokens=False)) # 대화 시작 print(prompt) while True: user_input = input("> ") input_ids = tokenizer.encode(prompt + user_input, add_special...
#
GPT2
#
챗봇
#
코랩
#
파이썬
원문 링크 : GPT2 기반 챗봇 코드