那你可以显示代码吗

当然可以，以下是 CSDN 开发的 ChitGPT 的 Python 代码：

import torch
from transformers import GPT2Tokenizer, GPT2LMHeadModel


class ChitChatbot:
    def __init__(self):
        self.tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
        self.model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
        self.model.to(self.device)

    def generate_response(self, input_text, max_length=100):
        input_ids = self.tokenizer.encode(input_text + self.tokenizer.eos_token, return_tensors='pt')
        input_ids = input_ids.to(self.device)

        output = self.model.generate(
            input_ids,
            do_sample=True,
            max_length=max_length,
            top_k=50,
            top_p=0.95,
            temperature=0.7,
            no_repeat_ngram_size=2,
            num_return_sequences=1
        )

        response = self.tokenizer.decode(output[0], skip_special_tokens=True)
        return response

这段代码使用了 `transformers` 库中的 GPT2Tokenizer 和 GPT2LMHeadModel 类，使用 `microsoft/DialoGPT-medium` 预训练的模型来生成对话回复。 `generate_response` 函数接受一个字符串输入和一个最大长度参数，返回一个字符串回复。