Updation for dlrm model and dataset download enable condition

anandhu-eng · Oct 1, 2024 · 09bc50a · 09bc50a
1 parent 332ec61
commit 09bc50a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/script/app-mlperf-inference-nvidia/_cm.yaml b/script/app-mlperf-inference-nvidia/_cm.yaml
@@ -162,9 +162,11 @@ deps:
       - dlrm-v2-99.9
     skip_if_any_env:
       DLRM_DATA_PATH:
-      - 'on'
+      - on
       CM_RUN_STATE_DOCKER:
       - 'yes'
+      - True
+      - 'True'
     names:
       - criteo-preprocessed
     tags: get,dataset,preprocessed,criteo
@@ -180,6 +182,8 @@ deps:
       - on
       CM_RUN_STATE_DOCKER:
       - 'yes'
+      - True
+      - 'True'
     names:
       - dlrm-model
     tags: get,ml-model,dlrm,_pytorch