update test marker

Author: faaany

src/diffusers/utils/testing_utils.py

@@ -337,6 +337,14 @@ def require_note_seq(test_case):
     return unittest.skipUnless(is_note_seq_available(), "test requires note_seq")(test_case)
 
 
+def require_non_cpu(test_case):
+    """
+    Decorator marking a test that requires a hardware accelerator backend. These tests are skipped when there are no
+    hardware accelerator available.
+    """
+    return unittest.skipUnless(torch_device != "cpu", "test requires a GPU")(test_case)
+
+
 def require_torchsde(test_case):
     """
     Decorator marking a test that requires torchsde. These tests are skipped when torchsde isn't installed.

tests/pipelines/animatediff/test_animatediff.py

@@ -19,7 +19,13 @@
 )
 from diffusers.models.attention import FreeNoiseTransformerBlock
 from diffusers.utils import is_xformers_available, logging
-from diffusers.utils.testing_utils import numpy_cosine_similarity_distance, require_torch_gpu, slow, torch_device
+from diffusers.utils.testing_utils import (
+    numpy_cosine_similarity_distance,
+    require_non_cpu,
+    require_torch_gpu,
+    slow,
+    torch_device,
+)
 
 from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_PARAMS
 from ..test_pipelines_common import (
@@ -272,7 +278,7 @@ def test_inference_batch_single_identical(
         max_diff = np.abs(to_np(output_batch[0][0]) - to_np(output[0][0])).max()
         assert max_diff < expected_max_diff
 
-    @unittest.skipIf(torch_device = "cpu", reason="Hardware Accelerator and CPU are required to switch devices")
+    @require_non_cpu
     def test_to_device(self):
         components = self.get_dummy_components()
         pipe = self.pipeline_class(**components)

tests/pipelines/latent_diffusion/test_latent_diffusion_superresolution.py

@@ -26,6 +26,7 @@
     floats_tensor,
     load_image,
     nightly,
+    require_non_cpu,
     require_torch,
     torch_device,
 )
@@ -93,7 +94,7 @@ def test_inference_superresolution(self):
 
         assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2
 
-    @unittest.skipIf(torch_device = "cpu", "This test requires a hardware accelerator")
+    @require_non_cpu
     def test_inference_superresolution_fp16(self):
         unet = self.dummy_uncond_unet
         scheduler = DDIMScheduler()

tests/pipelines/test_pipelines_common.py

@@ -771,7 +771,7 @@ def test_from_pipe_consistent_forward_pass(self, expected_max_diff=1e-3):
                 ), "`from_pipe` changed the attention processor in original pipeline."
 
     @unittest.skipIf(
-        torch_device = "cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
+        torch_device="cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
         reason="CPU offload is only available with hardware accelerator and `accelerate v0.14.0` or higher",
     )
     def test_from_pipe_consistent_forward_pass_cpu_offload(self, expected_max_diff=1e-3):
@@ -1317,7 +1317,7 @@ def test_save_load_optional_components(self, expected_max_difference=1e-4):
         max_diff = np.abs(to_np(output) - to_np(output_loaded)).max()
         self.assertLess(max_diff, expected_max_difference)
 
-    @unittest.skipIf(torch_device = "cpu", reason="Hardware accelerator and CPU are required to switch devices")
+    @unittest.skipIf(torch_device="cpu", reason="Hardware accelerator and CPU are required to switch devices")
     def test_to_device(self):
         components = self.get_dummy_components()
         pipe = self.pipeline_class(**components)
@@ -1392,7 +1392,7 @@ def _test_attention_slicing_forward_pass(
             assert_mean_pixel_difference(to_np(output_with_slicing2[0]), to_np(output_without_slicing[0]))
 
     @unittest.skipIf(
-        torch_device = "cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
+        torch_device="cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
         reason="CPU offload is only available with hardware accelerator and `accelerate v0.14.0` or higher",
     )
     def test_sequential_cpu_offload_forward_pass(self, expected_max_diff=1e-4):
@@ -1455,7 +1455,7 @@ def test_sequential_cpu_offload_forward_pass(self, expected_max_diff=1e-4):
         )
 
     @unittest.skipIf(
-        torch_device = "cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.17.0"),
+        torch_device="cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.17.0"),
         reason="CPU offload is only available with hardware accelerator and `accelerate v0.17.0` or higher",
     )
     def test_model_cpu_offload_forward_pass(self, expected_max_diff=2e-4):
@@ -1512,7 +1512,7 @@ def test_model_cpu_offload_forward_pass(self, expected_max_diff=2e-4):
         )
 
     @unittest.skipIf(
-        torch_device = "cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.17.0"),
+        torch_device="cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.17.0"),
         reason="CPU offload is only available with hardware accelerator and `accelerate v0.17.0` or higher",
     )
     def test_cpu_offload_forward_pass_twice(self, expected_max_diff=2e-4):
@@ -1569,7 +1569,7 @@ def test_cpu_offload_forward_pass_twice(self, expected_max_diff=2e-4):
         )
 
     @unittest.skipIf(
-        torch_device = "cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
+        torch_device="cpu" or not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
         reason="CPU offload is only available with hardware accelerator and `accelerate v0.14.0` or higher",
     )
     def test_sequential_offload_forward_pass_twice(self, expected_max_diff=2e-4):