进入本项目目录,训练启动命令如下:
python finetune.py data ../../../Model/THUDM/glm-4-9b-chat configs/lora.yaml
从保存检查点继续训练,命令如下:
python finetune.py data ../../../Model/THUDM/glm-4-9b-chat configs/lora.yaml yes
需要使用conda安装mpi4py
conda install mpi4py
OMP_NUM_THREADS=1 torchrun --standalone --nnodes=1 --nproc_per_node=8 finetune.py data ../../../Model/THUDM/glm-4-9b-chat configs/sft.yaml