你能帮我搭建一个示例吗

当然可以，以下是一个基于ChitGPT的简单示例：

import torch
from transformers import GPT2Tokenizer, GPT2LMHeadModel

tokenizer = GPT2Tokenizer.from_pretrained('chitgpt-small')
model = GPT2LMHeadModel.from_pretrained('chitgpt-small')

def generate_response(input_text, max_length=100):
    input_ids = tokenizer.encode(input_text, return_tensors='pt')
    output = model.generate(input_ids=input_ids, max_length=max_length, num_beams=5, no_repeat_ngram_size=2, early_stopping=True)
    response = tokenizer.decode(output[0], skip_special_tokens=True)
    return response

while True:
    user_input = input("You: ")
    response = generate_response(user_input)
    print("ChitGPT: " + response)

这个示例可以与用户进行简单的聊天，你可以根据自己的需求进行修改和扩展。