From b1e6d2d79dcb1728092f4f734528ffdeffc9dad1 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Thu, 23 Mar 2023 11:28:29 +0000 Subject: [PATCH 1/3] Skip mps in text-to-video tests. --- tests/pipelines/text_to_video/test_text_to_video.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/text_to_video/test_text_to_video.py b/tests/pipelines/text_to_video/test_text_to_video.py index eb43a360653a..c3dd0f63189d 100644 --- a/tests/pipelines/text_to_video/test_text_to_video.py +++ b/tests/pipelines/text_to_video/test_text_to_video.py @@ -34,7 +34,7 @@ torch.backends.cuda.matmul.allow_tf32 = False - +@skip_mps class TextToVideoSDPipelineFastTests(PipelineTesterMixin, unittest.TestCase): pipeline_class = TextToVideoSDPipeline params = TEXT_TO_IMAGE_PARAMS @@ -155,12 +155,12 @@ def test_inference_batch_single_identical(self): def test_num_images_per_prompt(self): pass - @skip_mps def test_progress_bar(self): return super().test_progress_bar() @slow +@skip_mps class TextToVideoSDPipelineSlowTests(unittest.TestCase): def test_full_model(self): expected_video = load_numpy( From c0a02b89a9f884bb06efb203071aa084dd1c522f Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Thu, 23 Mar 2023 11:31:13 +0000 Subject: [PATCH 2/3] style --- tests/pipelines/text_to_video/test_text_to_video.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/pipelines/text_to_video/test_text_to_video.py b/tests/pipelines/text_to_video/test_text_to_video.py index c3dd0f63189d..e4331fda02ff 100644 --- a/tests/pipelines/text_to_video/test_text_to_video.py +++ b/tests/pipelines/text_to_video/test_text_to_video.py @@ -34,6 +34,7 @@ torch.backends.cuda.matmul.allow_tf32 = False + @skip_mps class TextToVideoSDPipelineFastTests(PipelineTesterMixin, unittest.TestCase): pipeline_class = TextToVideoSDPipeline From 12339cf61c6aa51a8d71101673cc592bbbf73c6d Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Thu, 23 Mar 2023 12:07:08 +0000 Subject: [PATCH 3/3] Skip UNet3D mps tests. --- tests/models/test_models_unet_3d_condition.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/models/test_models_unet_3d_condition.py b/tests/models/test_models_unet_3d_condition.py index a92b8edd5378..ea71ae4af26c 100644 --- a/tests/models/test_models_unet_3d_condition.py +++ b/tests/models/test_models_unet_3d_condition.py @@ -23,6 +23,7 @@ from diffusers.utils import ( floats_tensor, logging, + skip_mps, torch_device, ) from diffusers.utils.import_utils import is_xformers_available @@ -60,6 +61,7 @@ def create_lora_layers(model): return lora_attn_procs +@skip_mps class UNet3DConditionModelTests(ModelTesterMixin, unittest.TestCase): model_class = UNet3DConditionModel