Force CPU in 2nd test

huggingface · Aug 11, 2023 · 01723c3 · 01723c3
1 parent e0c0ff8
commit 01723c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_accelerator.py b/tests/test_accelerator.py
@@ -361,7 +361,7 @@ def test_can_unwrap_model_fp16(self):
 
     def test_can_unwrap_model(self):
         model = create_components()[0]
-        accelerator = Accelerator(mixed_precision="no")
+        accelerator = Accelerator(mixed_precision="no", cpu=True)
         inputs = torch.randn(10, 2)
         model = accelerator.prepare(model)
         model(inputs)  # sanity check that this works