mixed

mosaicml · Feb 6, 2024 · b7884de · b7884de
1 parent 5ba5e30
commit b7884de
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}