add case

Author: faaany

tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py

@@ -30,7 +30,7 @@
     load_numpy,
     nightly,
     numpy_cosine_similarity_distance,
-    require_torch_gpu,
+    require_torch_accelerator,
     skip_mps,
     torch_device,
 )
@@ -205,7 +205,7 @@ def test_from_pipe_consistent_forward_pass_cpu_offload(self):
         super().test_from_pipe_consistent_forward_pass_cpu_offload(expected_max_diff=5e-3)
 
 
-@require_torch_gpu
+@require_torch_accelerator
 @nightly
 class StableDiffusionAttendAndExcitePipelineIntegrationTests(unittest.TestCase):
     # Attend and excite requires being able to run a backward pass at
@@ -237,7 +237,7 @@ def test_attend_and_excite_fp16(self):
         pipe = StableDiffusionAttendAndExcitePipeline.from_pretrained(
             "CompVis/stable-diffusion-v1-4", safety_checker=None, torch_dtype=torch.float16
         )
-        pipe.to("cuda")
+        pipe.to(torch_device)
 
         prompt = "a painting of an elephant with glasses"
         token_indices = [5, 7]