VS Code 中使用 ChatGPT 进行对话：详细步骤及代码示例

安装 Python 和 PyTorch 在使用 ChatGPT 前，需要先安装 Python 和 PyTorch。可以从 Python 官网下载 Python，从 PyTorch 官网下载 PyTorch。
安装 Transformers 库将 Transformers 库安装到 Python 环境中。打开终端窗口，输入以下命令：

pip install transformers

创建 Python 文件在 VS Code 中创建一个 Python 文件。在文件中导入必要的库：

from transformers import AutoTokenizer, AutoModelWithLMHead

tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-medium')
model = AutoModelWithLMHead.from_pretrained('microsoft/DialoGPT-medium')

使用 ChatGPT 进行对话使用以下代码进行对话：

# 记录对话历史
history = []

# 开始对话
while True:
    # 获取用户输入
    user_input = input('User: ')

    # 添加用户输入到历史记录中
    history.append(user_input)

    # 将历史记录转换为聊天格式
    chat_history = ' '.join(history)

    # 将聊天格式编码为模型输入
    input_ids = tokenizer.encode(chat_history, return_tensors='pt')

    # 生成响应
    response_ids = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)

    # 将响应解码为文本
    response = tokenizer.decode(response_ids[0], skip_special_tokens=True)

    # 打印响应
    print('ChatGPT: ' + response)

    # 添加响应到历史记录中
    history.append(response)

运行 Python 文件在 VS Code 中运行 Python 文件，输入任何对话内容即可开始与 ChatGPT 进行对话。