로딩
요청 처리 중입니다...

GPT2 기반 챗봇 코드

 GPT2 기반 챗봇 코드

에러 도저히 못 고칠거 같아서 그냥 올린다. 아무나 고칠 수 있는 사람은 가져가서 활용해주세요.. 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 # 챗봇 # 코랩 # 파이썬