代码如何
时间: 2024-01-09 13:03:56 浏览: 23
以下是使用Python编写的一个简单的ChitGPT对话AI代码示例:
```python
import torch
from transformers import GPT2LMHeadModel, GPT2Tokenizer
# 加载模型和分词器
tokenizer = GPT2Tokenizer.from_pretrained("microsoft/DialoGPT-small")
model = GPT2LMHeadModel.from_pretrained("microsoft/DialoGPT-small")
# 设置模型为评估模式
model.eval()
# 初始对话
print("ChitGPT: 你好,欢迎来到 ChitGPT 对话系统!")
while True:
# 接收用户输入
user_input = input("User: ")
# 把用户输入转换为模型可接受的输入格式
input_ids = tokenizer.encode(user_input, return_tensors="pt")
# 生成回复
output = model.generate(input_ids, max_length=50, pad_token_id=tokenizer.eos_token_id)
# 把模型生成的输出转换为可读的文本
response = tokenizer.decode(output[0], skip_special_tokens=True)
# 输出回复
print("ChitGPT: " + response)
```
这个示例代码使用了Hugging Face的`transformers`库,通过加载微软的DialoGPT-small模型和相应的分词器,实现了一个简单的对话系统。可以通过修改模型和分词器的名称来使用其他预训练的GPT模型,从而实现更加复杂的对话系统。