Update mcli-hf-eval.yaml (#411)

* Update mcli-hf-eval.yaml fix init device * Update mcli-hf-eval.yaml * updt --------- Co-authored-by: Vitaliy Chiley <[email protected]>
mosaicml · Jul 1, 2023 · 29d65cc · 29d65cc
1 parent 5c14661
commit 29d65cc
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/mcli/mcli-hf-eval.yaml b/mcli/mcli-hf-eval.yaml
@@ -38,7 +38,7 @@ parameters:
     model:
       name: hf_causal_lm
       pretrained_model_name_or_path: mosaicml/mpt-7b-instruct
-      device: cpu
+      init_device: cpu
       pretrained: true
       use_auth_token: false
   # -
@@ -52,7 +52,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: tiiuae/falcon-40b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -66,7 +66,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: mosaicml/mpt-30b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: true
   # -
@@ -80,7 +80,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: tiiuae/falcon-40b-instruct
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -94,7 +94,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: huggyllama/llama-30b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -108,7 +108,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: mosaicml/mpt-7b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -122,7 +122,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: mosaicml/mpt-7b-chat
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -136,7 +136,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: mosaicml/mpt-7b-instruct
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -150,7 +150,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: tiiuae/falcon-7b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -164,7 +164,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: togethercomputer/RedPajama-INCITE-7B-Instruct
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -178,7 +178,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: togethercomputer/RedPajama-INCITE-7B-Base
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -192,7 +192,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: huggyllama/llama-13b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -206,7 +206,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: huggyllama/llama-7b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -220,7 +220,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: EleutherAI/pythia-12b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -234,7 +234,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: EleutherAI/pythia-6.9b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -248,7 +248,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: facebook/opt-6.7b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -262,7 +262,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: stabilityai/stablelm-tuned-alpha-7b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false
   # -
@@ -276,7 +276,7 @@ parameters:
   #   model:
   #     name: hf_causal_lm
   #     pretrained_model_name_or_path: EleutherAI/gpt-j-6b
-  #     device: cpu
+  #     init_device: cpu
   #     pretrained: true
   #     use_auth_token: false