Fixes for intel-gptj

ctuning · Feb 10, 2024 · 4b390fc · 4b390fc
1 parent e85ec51
commit 4b390fc
Show file tree

Hide file tree

Showing 2 changed files with 3 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
@@ -181,7 +181,6 @@
     },
     "wget": {
       "group": "download-tool",
-      "default": true,
       "add_deps_recursive": {
         "dae": {
           "tags": "_wget"
@@ -194,6 +193,7 @@
     },
     "rclone": {
       "group": "download-tool",
+      "default": true,
       "add_deps_recursive": {
         "dae": {
           "tags": "_rclone"

diff --git a/cm-mlops/script/reproduce-mlperf-inference-intel/_cm.yaml b/cm-mlops/script/reproduce-mlperf-inference-intel/_cm.yaml
@@ -295,9 +295,11 @@ variations:
       - tags: get,generic-python-lib,_package.datasets
         names:
           - pip-package
+          - datasets
       - tags: get,generic-python-lib,_package.accelerate
         names:
           - pip-package
+          - accelerate
 
   gptj-99:
     group: model