Skip to content

Commit 9f9f943

Browse files
guangy10malfet
authored andcommitted
Fix download script location (#379)
1 parent e96fce0 commit 9f9f943

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

scripts/workflow.sh

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)