Skip to content

GitLab

Commit 412055be authored by dengjb's avatar dengjb
Browse files

Update train.sh

parent 225b3e00
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train.sh
View file @ 412055be
...@@ -6,7 +6,7 @@ MODEL_PATH="./weights/CodeLlama-7b-Instruct-hf" ...@@ -6,7 +6,7 @@ MODEL_PATH="./weights/CodeLlama-7b-Instruct-hf"
DEEPSEPPD_CONFIG="./dp_configs/ds_config_stage3.json" DEEPSEPPD_CONFIG="./dp_configs/ds_config_stage3.json"
LORA_CONFIG="./lora_config.json" LORA_CONFIG="./lora_config.json"
deepspeed --include="localhost:2,3" --master_port 25923 fintune.py \ deepspeed --include="localhost:2,3" --master_port 25923 finetune.py \
--model_name_or_path $MODEL_PATH \ --model_name_or_path $MODEL_PATH \
--data_path $DATA_PATH \ --data_path $DATA_PATH \
--output_dir $OUTPUT_PATH \ --output_dir $OUTPUT_PATH \
......
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