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() {
51
51
if [ " $FORCE_DOWNLOAD " = true ] || [ ! -d " $CHECKPOINT_DIR " ] || [ -z " $( ls -A " $CHECKPOINT_DIR " ) " ]; then
52
52
echo " Download checkpoint for $MODEL_REPO "
53
53
rm -rf " $CHECKPOINT_DIR "
54
- python3 scripts/ download.py --repo-id " $MODEL_REPO "
54
+ python3 download.py --repo-id " $MODEL_REPO "
55
55
else
56
56
echo " Checkpoint directory for $MODEL_REPO is not empty. Skipping download."
57
57
fi
@@ -62,7 +62,17 @@ function run_validation_e2e() {
62
62
63
63
echo " "
64
64
echo " ############### Validating ${MODEL_REPO##*/ } ###############"
65
+
66
+ if [ ! -f " download.py" ]; then
67
+ echo " download.py doesn't exist."
68
+ exit 1
69
+ fi
65
70
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
66
76
bash .ci/scripts/convert_checkpoint.sh " $MODEL_REPO "
67
77
68
78
set +e
You can’t perform that action at this time.
0 commit comments