almost missed you

Author: SalmanMohammadi

tests/recipes/test_eleuther_eval.py

@@ -232,7 +232,7 @@ def test_meta_eval_vision(self, caplog, monkeypatch, tmpdir, expected_vision_acc
             tokenizer.prompt_template=null \
             limit=1 \
             dtype=bf16 \
-            device=cuda \
+            device=cpu \
         """.split()
 
         model_config = llama3_2_vision_test_config()
@@ -274,7 +274,7 @@ def test_hf_eval_vision(self, caplog, monkeypatch, tmpdir, expected_vision_acc):
             tokenizer.prompt_template=null \
             limit=1 \
             dtype=bf16 \
-            device=cuda \
+            device=cpu \
         """.split()
 
         model_config = llama3_2_vision_test_config()