File tree Expand file tree Collapse file tree 1 file changed +11
-1
lines changed Expand file tree Collapse file tree 1 file changed +11
-1
lines changed Original file line number Diff line number Diff line change @@ -51,7 +51,7 @@ function download_checkpoint() {
5151 if [ " $FORCE_DOWNLOAD " = true ] || [ ! -d " $CHECKPOINT_DIR " ] || [ -z " $( ls -A " $CHECKPOINT_DIR " ) " ]; then
5252 echo " Download checkpoint for $MODEL_REPO "
5353 rm -rf " $CHECKPOINT_DIR "
54- python3 scripts/ download.py --repo-id " $MODEL_REPO "
54+ python3 download.py --repo-id " $MODEL_REPO "
5555 else
5656 echo " Checkpoint directory for $MODEL_REPO is not empty. Skipping download."
5757 fi
@@ -62,7 +62,17 @@ function run_validation_e2e() {
6262
6363 echo " "
6464 echo " ############### Validating ${MODEL_REPO##*/ } ###############"
65+
66+ if [ ! -f " download.py" ]; then
67+ echo " download.py doesn't exist."
68+ exit 1
69+ fi
6570 download_checkpoint " $MODEL_REPO "
71+
72+ if [ ! -f " build/convert_hf_checkpoint.py" ]; then
73+ echo " build/convert_hf_checkpoint.py doesn't exist."
74+ exit 1
75+ fi
6676 bash .ci/scripts/convert_checkpoint.sh " $MODEL_REPO "
6777
6878 set +e
You can’t perform that action at this time.
0 commit comments