Update hf_eval.yaml

mosaicml · Feb 2, 2024 · c6f2109 · c6f2109
1 parent 59f1de0
commit c6f2109
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/eval/yamls/hf_eval.yaml b/scripts/eval/yamls/hf_eval.yaml
@@ -12,7 +12,7 @@ models:
   model:
     name: hf_causal_lm
     pretrained_model_name_or_path: ${model_name_or_path}
-    init_device: cpu
+    init_device: mixed
     pretrained: true
   tokenizer:
     name: ${model_name_or_path}