如何在python上使用chat gpt

要在 Python 上使用 ChatGPT，可以使用 Hugging Face Transformers 库，它提供了许多预先训练好的自然语言处理模型，包括 ChatGPT。

以下是使用 Hugging Face Transformers 库在 Python 上使用 ChatGPT 的示例代码：

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

# 加载 ChatGPT 模型和分词器
model_name = "microsoft/DialoGPT-medium"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

# 设置对话历史和生成新的回复
conversation_history = "Hello, how are you today?"
new_input = "I'm doing well, thanks for asking. How about you?"

# 将对话历史和新的输入编码成模型输入格式
input_ids = tokenizer.encode(conversation_history + tokenizer.eos_token + new_input + tokenizer.eos_token, return_tensors='pt')

# 生成新的回复
bot_output = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)

# 将生成的回复解码成人类可读的字符串形式
bot_reply = tokenizer.decode(bot_output[0], skip_special_tokens=True)

print("Bot reply:", bot_reply)

这个示例代码加载了 Microsoft's DialoGPT-medium 模型，并使用分词器将对话历史和新的输入编码成输入格式。然后，它使用模型的 generate 方法生成新的回复，并使用分词器将回复解码为人类可读的字符串形式。您可以根据需要修改对话历史和新的输入。