给个示例代码

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

from chitgpt import ChitGPT
import torch

# 加载预训练模型
model = ChitGPT.from_pretrained('chitgpt-medium')

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

# 设置设备为CPU或GPU
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

# 将模型移动到设备上
model.to(device)

while True:
    # 获取用户输入
    user_input = input('你： ').strip()

    # 如果用户输入为空，则继续等待用户输入
    if not user_input:
        continue

    # 将用户输入转换为模型输入的编码
    inputs = model.tokenizer(user_input, return_tensors='pt').input_ids.to(device)

    # 使用模型生成回复
    outputs = model.generate(inputs, max_length=50, do_sample=True)

    # 将模型生成的回复转换为可读的字符串
    reply = model.tokenizer.decode(outputs[0], skip_special_tokens=True)

    # 输出模型生成的回复
    print('ChitGPT：', reply)

在上面的代码中，我们首先加载了预训练的ChitGPT模型，并将其设置为eval模式。然后，我们使用一个while循环来不断等待用户输入，并使用模型生成回复。最后，我们将生成的回复输出到屏幕上。