Commit e9bec754 authored by zhaoying1's avatar zhaoying1
Browse files

Update run_train_single.sh

parent 9f69dd10
......@@ -24,8 +24,8 @@ APP="python3 ../main.py \
--deepspeed ../deepspeed.json \
--do_train \
--train_file AdvertiseGen/train.json \
--prompt_column prompt \
--response_column response \
--prompt_column content \
--response_column summary \
--model_name_or_path THUDM/chatglm-6b \
--output_dir ./output_ft/pretrain \
--overwrite_output_dir \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment