import torch from transformers import AutoTokenizer, AutoModelForCausalLM # 加載預(yù)訓(xùn)練的GPT-2模型和tokenizer model_name = "microsoft/DialoGPT-large" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForCausalLM.from_pretrained(model_name) # 輸入對話文本，生成回復(fù) def generate_response(prompt): input_ids = tokenizer.encode(prompt + tokenizer.eos_token, return_tensors='pt') output = model.generate(input_ids=input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id) response = tokenizer.decode(output[0], skip_special_tokens=True) return response