skip model and dataset download if env variable is set

anandhu-eng · Sep 30, 2024 · 8d942a8 · 8d942a8
1 parent 20a47af
commit 8d942a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/script/app-mlperf-inference-nvidia/_cm.yaml b/script/app-mlperf-inference-nvidia/_cm.yaml
@@ -165,6 +165,8 @@ deps:
       - on
       CM_RUN_STATE_DOCKER:
       - 'yes'
+      CM_MLPERF_MODEL_DLRM_SKIP_DOWNLOAD:
+      - 'yes'
     names:
       - criteo-preprocessed
     tags: get,dataset,preprocessed,criteo
@@ -180,6 +182,8 @@ deps:
       - on
       CM_RUN_STATE_DOCKER:
       - 'yes'
+      CM_MLPERF_MODEL_DLRM_SKIP_DOWNLOAD:
+      - 'yes'
     names:
       - dlrm-model
     tags: get,ml-model,dlrm,_pytorch