From aff3c93c7653df07176f5c20cb7ab2fd5b6b987f Mon Sep 17 00:00:00 2001 From: ddeellttaa <49911567+ddeellttaa@users.noreply.github.com> Date: Sat, 15 Jun 2024 14:12:15 +0800 Subject: [PATCH] Update README.md change finetune_hf.py to finetune.py --- finetune_demo/README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/finetune_demo/README.md b/finetune_demo/README.md index 83342d2..022c2f3 100644 --- a/finetune_demo/README.md +++ b/finetune_demo/README.md @@ -166,7 +166,7 @@ OMP_NUM_THREADS=1 torchrun --standalone --nnodes=1 --nproc_per_node=8 finetune_ 通过以下代码执行 **单机单卡** 运行。 ```shell -python finetune_hf.py data/AdvertiseGen/ THUDM/glm-4-9b-chat configs/lora.yaml +python finetune.py data/AdvertiseGen/ THUDM/glm-4-9b-chat configs/lora.yaml ``` ## 从保存点进行微调 @@ -179,7 +179,7 @@ python finetune_hf.py data/AdvertiseGen/ THUDM/glm-4-9b-chat configs/lora.yam 例如,这就是一个从最后一个保存点继续微调的示例代码 ```shell -python finetune_hf.py data/AdvertiseGen/ THUDM/glm-4-9b-chat configs/lora.yaml yes +python finetune.py data/AdvertiseGen/ THUDM/glm-4-9b-chat configs/lora.yaml yes ``` ## 使用微调后的模型