Prevent error in rclone detect

mlcommons · Feb 2, 2024 · 31d64f3 · 31d64f3
1 parent e5619d2
commit 31d64f3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cm-mlops/script/get-ml-model-gptj/_cm.json b/cm-mlops/script/get-ml-model-gptj/_cm.json
@@ -19,6 +19,7 @@
   "prehook_deps": [
     {
       "env": {
+        "CM_DOWNLOAD_FINAL_ENV_NAME": "GPTJ_CHECKPOINT_PATH",
         "CM_EXTRACT_FINAL_ENV_NAME": "GPTJ_CHECKPOINT_PATH",
         "CM_EXTRACT_TO_FOLDER": "gpt-j"
       },
@@ -87,7 +88,9 @@
         "CM_PACKAGE_URL": "https://cloud.mlcommons.org/index.php/s/QAZ2oM94MkFtbQx/download",
         "CM_RCLONE_CONFIG_CMD": "rclone config create mlc-inference s3 provider=LyveCloud access_key_id=0LITLNQMHZALM5AK secret_access_key=YQKYTMBY23TMZHLOYFJKL5CHHS0CWYUC endpoint=s3.us-east-1.lyvecloud.seagate.com",
         "CM_RCLONE_URL": "mlc-inference:mlcommons-inference-wg-s3/gpt-j"
-      },
+      }
+    },
+    "pytorch,fp32,wget": {
       "add_deps_recursive": {
         "dae": {
           "tags": "_extract"