enable

Author: faaany

tests/pipelines/stable_video_diffusion/test_stable_video_diffusion.py

@@ -403,8 +403,8 @@ def test_to_dtype(self):
         self.assertTrue(all(dtype == torch.float16 for dtype in model_dtypes))
 
     @unittest.skipIf(
-        torch_device != "cuda" or not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
-        reason="CPU offload is only available with CUDA and `accelerate v0.14.0` or higher",
+        not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
+        reason="CPU offload is only available with `accelerate v0.14.0` or higher",
     )
     def test_sequential_cpu_offload_forward_pass(self, expected_max_diff=1e-4):
         components = self.get_dummy_components()
@@ -419,7 +419,7 @@ def test_sequential_cpu_offload_forward_pass(self, expected_max_diff=1e-4):
         inputs = self.get_dummy_inputs(generator_device)
         output_without_offload = pipe(**inputs).frames[0]
 
-        pipe.enable_sequential_cpu_offload()
+        pipe.enable_sequential_cpu_offload(device=torch_device)
 
         inputs = self.get_dummy_inputs(generator_device)
         output_with_offload = pipe(**inputs).frames[0]