Skip to content

Commit d4b0238

Browse files
committed
Fix abtf-model download
1 parent 3579623 commit d4b0238

3 files changed

Lines changed: 9 additions & 4 deletions

File tree

script/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# MLCommons Automation Scripts
22

3-
*Last updated: 2026-04-18 05:05:05*
3+
*Last updated: 2026-04-18 05:24:39*
44

55
This directory contains automation scripts for MLPerf benchmarks, AI/ML workflows, and development operations.
66

script/get-ml-model-abtf-ssd-pytorch/customize.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,10 @@ def postprocess(i):
4646
env['MLC_ML_MODEL_SSD_PATH'], env['MLC_ML_MODEL_FILENAME'])
4747
env['MLC_ML_MODEL_FILE_WITH_PATH'] = env['MLC_ML_MODEL_SSD_PATH']
4848

49+
else:
50+
if env.get('MLC_ML_MODEL_SSD_PATH', '') == '':
51+
env['MLC_ML_MODEL_SSD_PATH'] = env['MLC_ML_MODEL_FILE_WITH_PATH']
52+
4953
env['MLC_ML_MODEL_FILE'] = os.path.basename(
5054
env['MLC_ML_MODEL_FILE_WITH_PATH'])
5155

script/get-ml-model-abtf-ssd-pytorch/meta.yaml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -95,10 +95,11 @@ variations:
9595
gdown:
9696
group: download-tool
9797
env:
98-
MLC_DOWNLOAD_EXTRA_OPTIONS: ' --fuzzy'
98+
MLC_DOWNLOAD_EXTRA_OPTIONS: ' '
9999
adr:
100100
abtf-ml-model-weights-download:
101101
tags: _gdown
102+
102103
r2-downloader:
103104
group: download-tool
104105
rclone:
@@ -159,7 +160,7 @@ variations:
159160
group: model-weights
160161
abtf-poc:
161162
default_variations:
162-
download-tool: rclone
163+
download-tool: gdown
163164
env:
164165
MLC_ML_MODEL: abtf-ssd-pytorch
165166
MLC_ML_MODEL_CHECKSUM: 26845c3b9573ce115ef29dca4ae5be14
@@ -202,7 +203,7 @@ variations:
202203
group: run-mode
203204
env:
204205
MLC_DOWNLOAD_MODE: dry
205-
abtf-poc,gdrive:
206+
abtf-poc,gdown:
206207
env:
207208
MLC_ML_MODEL_URL: https://drive.google.com/file/d/1kfJR_bs54KONprVd51kZu0PYmmh1wZZa/view
208209
abtf-poc,rclone:

0 commit comments

Comments
 (0)