Python chatting

import torch from transformers import AutoTokenizer, AutoModelForCausalLM # 加载预训练的GPT-2模型和tokenizer model_name = "microsoft/DialoGPT-large" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForCausalLM.from_pretrained(model_name) # 输入对话文本，生成回复 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