有没有示例代码

以下是一个使用ChitGPT进行对话的示例代码：

from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

tokenizer = AutoTokenizer.from_pretrained("chitgpt-medium")
model = AutoModelForCausalLM.from_pretrained("chitgpt-medium")

# 设置模型为eval模式
model.eval()

# 定义要与AI对话的文本
text = "你好，我是ChitGPT，我能帮你做什么？"

# 将文本编码成输入ids
input_ids = tokenizer.encode(text, return_tensors="pt")

# 生成回答
output = model.generate(input_ids=input_ids, max_length=50, do_sample=True)

# 将输出解码成文本并打印
response = tokenizer.decode(output[0], skip_special_tokens=True)
print(response)

此代码将使用ChitGPT-medium进行对话。您可以通过更改`"chitgpt-medium"`和`"chitgpt-medium"`为其他预训练模型来尝试不同的模型。

请注意，此示例代码仅用于演示如何使用ChitGPT进行对话。在实际应用中，需要考虑更多的输入处理、输出处理、对话历史记录、对话管理等问题。