Baize Chatbot 项目使用教程

1. 项目的目录结构及介绍

Baize Chatbot 项目的目录结构如下：

baize-chatbot/
├── data/
│   ├── Quora/
│   ├── StackOverFlow/
│   └── MedQuAD/
├── scripts/
│   ├── collect_data.py
│   ├── train.py
│   └── demo.py
├── models/
│   ├── baize-v1-7B/
│   ├── baize-v1-13B/
│   └── baize-v1-30B/
├── config/
│   ├── config.yaml
│   └── settings.json
├── README.md
└── requirements.txt

目录介绍

• data/: 包含从 Quora、StackOverFlow 和 MedQuAD 收集的对话数据。
• scripts/: 包含收集数据、训练模型和运行演示的脚本。
• models/: 包含不同版本的 Baize 模型。
• config/: 包含项目的配置文件。
• README.md: 项目说明文档。
• requirements.txt: 项目依赖包列表。

2. 项目的启动文件介绍

项目的启动文件主要是 scripts/demo.py，该文件用于启动 Baize Chatbot 的演示界面。

启动文件介绍

• demo.py: 该脚本负责加载训练好的模型并启动一个简单的 Web 界面，供用户与聊天机器人进行交互。

启动命令：

python scripts/demo.py

3. 项目的配置文件介绍

项目的配置文件位于 config/ 目录下，主要包括 config.yaml 和 settings.json。

配置文件介绍

• config.yaml: 包含模型的参数设置、数据路径和其他配置选项。
• settings.json: 包含项目的全局设置，如日志级别、端口号等。

配置文件示例

config.yaml 示例：

model:
  name: baize-v1-7B
  path: models/baize-v1-7B
data:
  path: data/Quora
  batch_size: 32

settings.json 示例：

{
  "log_level": "INFO",
  "port": 8080
}

通过以上配置文件，可以灵活地调整项目的运行参数和环境设置。