# 导入相关库from transformers import GPT2LMHeadModel, GPT2Tokenizer# 加载中文GPT模型和tokenizermodel = GPT2LMHeadModel.from_pretrained('clue/gpt2-base-chinese')tokenizer = GPT2Tokenizer.from_pretrained('clue/gpt2-base-chinese')# 设置输入文本text = "你好"# 将输入文本编码为token IDinput_ids = tokenizer.encode(text, return_tensors='pt')# 使用模型生成中文对话generated_text = model.generate( input_ids=input_ids, max_length=100, do_sample=True, temperature=0.7, top_k=50, top_p=0.95,)# 解码生成的token ID为可读的文本generated_text = tokenizer.decode(generated_text.squeeze(), skip_special_tokens=True)print(generated_text)
