Skip to content

Commit 8b5b3c5

Browse files
lekurileLeetJoe
authored andcommitted
Fix output arg in step 3 llama script (deepspeedai#668)
This PR fixes the OUTPUT arg in the step 3 llama script to use the proper arg position.
1 parent ee2b65b commit 8b5b3c5

File tree

1 file changed

+1
-1
lines changed
  • applications/DeepSpeed-Chat/training/step3_rlhf_finetuning/training_scripts/single_node

1 file changed

+1
-1
lines changed

applications/DeepSpeed-Chat/training/step3_rlhf_finetuning/training_scripts/single_node/run_7b_llama.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ ACTOR_MODEL_PATH=$1
77
CRITIC_MODEL_PATH=$2
88
ACTOR_ZERO_STAGE=$3
99
CRITIC_ZERO_STAGE=$4
10-
OUTPUT=$3
10+
OUTPUT=$5
1111
if [ "$OUTPUT" == "" ]; then
1212
OUTPUT=./output_step3_llama
1313
fi

0 commit comments

Comments
 (0)