ChatGLM-Instruct-Tuning

基于清华的 ChatGLM-6B + Alpaca 方式进行finetune.

数据集: 中文alpaca

准备

安装依赖

pip install -r requirements.txt

下载数据

cd data
git clone https://github.com/carbonz0/alpaca-chinese-dataset

数据预处理

转化alpaca数据集为按行存储的Intruct格式数据

python cover_alpaca2jsonl.py

然后把数据划分为train.txt和valid.txt，保存在 ./data/example/路径下

训练

bash scripts/finetune.sh

推理

# 要先把文件中的 "output/your_model_dir"替换为实际模型路径
python infer.py

交流

如果有疑问和建议，欢迎加入我们的大模型交流群