Support download sdxl model to host for nvidia implementation

anandhu-eng · Sep 19, 2024 · 88cc066 · 88cc066
1 parent 1fd7a44
commit 88cc066
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/script/app-mlperf-inference-nvidia/_cm.yaml b/script/app-mlperf-inference-nvidia/_cm.yaml
@@ -295,6 +295,11 @@ prehook_deps:
       - sdxl-model
       - ml-model
     tags: get,ml-model,sdxl,_fp16,_rclone
+    skip_if_env:
+      CM_RUN_STATE_DOCKER:
+      - 'yes'
+      CM_MLPERF_MODEL_SDXL_DOWNLOAD_TO_HOST:
+      - 'yes'
 
 # Post dependencies to run this app including for power measurement
 post_deps: