将这三个问题写入同一个代码
时间: 2024-01-24 14:17:12 浏览: 17
以下是一个使用 ChitGPT 进行多次对话的示例代码:
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
# 加载ChitGPT模型和分词器
model = AutoModelForCausalLM.from_pretrained("mymusise/ChitGPT")
tokenizer = AutoTokenizer.from_pretrained("mymusise/ChitGPT")
# 进行多次对话
while True:
# 读取用户输入
user_input = input("你: ")
# 将用户输入编码为模型所需格式
input_ids = tokenizer.encode(user_input, return_tensors="pt")
# 使用模型生成回复
output = model.generate(input_ids=input_ids, max_length=50, pad_token_id=tokenizer.eos_token_id)
# 解码模型生成的回复并打印出来
response = tokenizer.decode(output[0], skip_special_tokens=True)
print("ChitGPT: " + response)
# 如果用户输入 "退出",则退出对话
if user_input == "退出":
break
```
运行代码后,你可以尝试向 ChitGPT 提出三个问题,例如:
```
你: 你好,今天天气怎么样?
ChitGPT: 今天的天气还不错,有些凉爽。
你: 请问你是谁开发的?
ChitGPT: 我的开发者是CSDN。
你: 可以介绍一下你的功能吗?
ChitGPT: 我是一个类 ChatGPT 的 AI,可以用来进行对话。