请使用plantuml帮助我画出类图

好的，以下是使用PlantUML绘制的类图：

@startuml
class ChitGPT {
    -model: GPT2LMHeadModel
    -tokenizer: GPT2Tokenizer
    -device: str
    -temperature: float
    -top_p: float
    -max_length: int
    -min_length: int
    +__init__(model_dir: str)
    +generate_response(input_text: str) -> str
}

class GPT2LMHeadModel {
    # Attributes omitted for brevity
    +forward(input_ids, attention_mask, position_ids, head_mask) -> Tuple[torch.FloatTensor]
}

class GPT2Tokenizer {
    # Attributes omitted for brevity
    +encode(text: str, add_special_tokens: bool = True) -> List[int]
}

ChitGPT -> GPT2LMHeadModel
ChitGPT -> GPT2Tokenizer
@enduml

该类图展示了ChitGPT、GPT2LMHeadModel和GPT2Tokenizer之间的关系。其中，ChitGPT包含一个GPT2LMHeadModel和一个GPT2Tokenizer对象，可以使用它们来生成聊天响应。GPT2LMHeadModel和GPT2Tokenizer之间没有任何关系。