Skip to content

Commit

Permalink
because PaddleNLP has changed path of model_zoo, we also changed in c…
Browse files Browse the repository at this point in the history
…i_case_auto.sh and ci_case_dy.sh (#8633)
  • Loading branch information
jeff41404 committed Jun 20, 2024
1 parent da8b9ac commit 5619cc3
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion scripts/distribute/ci_case_auto.sh
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ set -e
export log_path=/workspace/case_logs
export root_path=/workspace/PaddleNLP

export gpt_case_path=$root_path/model_zoo/gpt-3
export gpt_case_path=$root_path/legacy/model_zoo/gpt-3
export gpt_data_path=/fleetx_data

export llama_case_path=$root_path/llm/llama/auto_parallel
Expand Down
6 changes: 3 additions & 3 deletions scripts/distribute/ci_case_dy.sh
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ set -e
export log_path=/workspace/case_logs
export root_path=/workspace/PaddleNLP

export gpt_case_path=$root_path/model_zoo/gpt-3
export gpt_case_path=$root_path/legacy/model_zoo/gpt-3
export gpt_data_path=/fleetx_data

export llm_gpt_case_path=$root_path/llm/gpt-3
Expand Down Expand Up @@ -273,7 +273,7 @@ function gpt_export_345M_mp1() {
rm -rf $log_dir
rm -rf output

export PYTHONPATH=$root_path/model_zoo/gpt-3:$PYTHONPATH
export PYTHONPATH=$root_path/legacy/model_zoo/gpt-3:$PYTHONPATH
export CUDA_VISIBLE_DEVICES=1
python -m paddle.distributed.launch --log_dir $log_dir --devices "1" \
./tools/auto_export.py \
Expand All @@ -294,7 +294,7 @@ function gpt_export_345M_mp2() {
rm -rf $log_dir
rm -rf output

export PYTHONPATH=$root_path/model_zoo/gpt-3:$PYTHONPATH
export PYTHONPATH=$root_path/legacy/model_zoo/gpt-3:$PYTHONPATH
export CUDA_VISIBLE_DEVICES=0,1
python -m paddle.distributed.launch --devices "0,1" \
./tools/auto_export.py \
Expand Down

0 comments on commit 5619cc3

Please sign in to comment.