github中显示的usage

对于ChitGPT的使用，可以参考以下步骤：

1. 安装依赖：

pip install torch transformers sentencepiece

2. 下载预训练模型：

git clone https://github.com/ChitraLakshmanan/ChitGPT.git

3. 加载模型：

from transformers import GPT2LMHeadModel, GPT2Tokenizer

model = GPT2LMHeadModel.from_pretrained("ChitraLakshmanan/ChitGPT")
tokenizer = GPT2Tokenizer.from_pretrained("ChitraLakshmanan/ChitGPT")

4. 生成文本：

input_text = "Hello, how are you?"
input_ids = tokenizer.encode(input_text, return_tensors='pt')
output = model.generate(input_ids, max_length=50, num_beams=5, no_repeat_ngram_size=2)
output_text = tokenizer.decode(output[0], skip_special_tokens=True)
print(output_text)

在这个例子中，我们生成了一个长度为50的文本，使用5个beam搜索，并且限制了不会重复出现2-gram。输出的文本可以通过tokenizer.decode方法进行解码。