Support download path for mlcommons large models

ctuning · Feb 28, 2024 · 8318d57 · 8318d57
1 parent 7d3b661
commit 8318d57
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/cm-mlops/script/get-ml-model-dlrm-terabyte/_cm.json b/cm-mlops/script/get-ml-model-dlrm-terabyte/_cm.json
@@ -16,7 +16,9 @@
     "CM_ML_MODEL_*"
   ],
   "input_mapping": {
-    "dir": "CM_DOWNLOAD_PATH"
+    "dir": "CM_DOWNLOAD_PATH",
+    "download_path": "CM_DOWNLOAD_PATH",
+    "to": "CM_DOWNLOAD_PATH"
   },
   "tags": [
     "get",

diff --git a/cm-mlops/script/get-ml-model-gptj/_cm.json b/cm-mlops/script/get-ml-model-gptj/_cm.json
@@ -10,7 +10,9 @@
     "CM_ML_MODEL_WEIGHT_TRANSFORMATIONS": "no"
   },
   "input_mapping": {
-    "checkpoint": "GPTJ_CHECKPOINT_PATH"
+    "checkpoint": "GPTJ_CHECKPOINT_PATH",
+    "download_path": "CM_DOWNLOAD_PATH",
+    "to": "CM_DOWNLOAD_PATH"
   },
   "new_env_keys": [
     "CM_ML_MODEL_*",

diff --git a/cm-mlops/script/get-ml-model-stable-diffusion/_cm.json b/cm-mlops/script/get-ml-model-stable-diffusion/_cm.json
@@ -11,7 +11,8 @@
   },
   "input_mapping": {
     "checkpoint": "SDXL_CHECKPOINT_PATH",
-    "download_path": "CM_DOWNLOAD_PATH"
+    "download_path": "CM_DOWNLOAD_PATH",
+    "to": "CM_DOWNLOAD_PATH"
   },
   "new_env_keys": [
     "CM_ML_MODEL_*",