Skip to content

Commit fff5954

Browse files
SunMarcSunMarc
committed
add class attributes to flag dduf tests
1 parent 7bc9347 commit fff5954

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

42 files changed

+93
-196
lines changed

‎tests/pipelines/audioldm/test_audioldm.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,8 @@ class AudioLDMPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
6363
]
6464
)
6565

66+
supports_dduf = False
67+
6668
def get_dummy_components(self):
6769
torch.manual_seed(0)
6870
unet = UNet2DConditionModel(
@@ -370,10 +372,6 @@ def test_inference_batch_single_identical(self):
370372
def test_xformers_attention_forwardGenerator_pass(self):
371373
self._test_xformers_attention_forwardGenerator_pass(test_mean_pixel_difference=False)
372374

373-
@unittest.skip("Test not supported.")
374-
def test_save_load_dduf(self):
375-
pass
376-
377375

378376
@nightly
379377
class AudioLDMPipelineSlowTests(unittest.TestCase):

‎tests/pipelines/audioldm2/test_audioldm2.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,8 @@ class AudioLDM2PipelineFastTests(PipelineTesterMixin, unittest.TestCase):
7070
]
7171
)
7272

73+
supports_dduf = False
74+
7375
def get_dummy_components(self):
7476
torch.manual_seed(0)
7577
unet = AudioLDM2UNet2DConditionModel(
@@ -509,10 +511,6 @@ def test_to_dtype(self):
509511
def test_sequential_cpu_offload_forward_pass(self):
510512
pass
511513

512-
@unittest.skip("Test not supported.")
513-
def test_save_load_dduf(self):
514-
pass
515-
516514

517515
@nightly
518516
class AudioLDM2PipelineSlowTests(unittest.TestCase):

‎tests/pipelines/blipdiffusion/test_blipdiffusion.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,8 @@ class BlipDiffusionPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
6060
"prompt_reps",
6161
]
6262

63+
supports_dduf = False
64+
6365
def get_dummy_components(self):
6466
torch.manual_seed(0)
6567
text_encoder_config = CLIPTextConfig(
@@ -196,7 +198,3 @@ def test_blipdiffusion(self):
196198
assert (
197199
np.abs(image_slice.flatten() - expected_slice).max() < 1e-2
198200
), f" expected_slice {image_slice.flatten()}, but got {image_slice.flatten()}"
199-
200-
@unittest.skip("Test not supported.")
201-
def test_save_load_dduf(self):
202-
pass

‎tests/pipelines/controlnet/test_controlnet.py‎

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -282,10 +282,6 @@ def test_controlnet_lcm_custom_timesteps(self):
282282

283283
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2
284284

285-
@unittest.skip("Test not supported.")
286-
def test_save_load_dduf(self):
287-
pass
288-
289285

290286
class StableDiffusionMultiControlNetPipelineFastTests(
291287
IPAdapterTesterMixin, PipelineTesterMixin, PipelineKarrasSchedulerTesterMixin, unittest.TestCase
@@ -295,6 +291,8 @@ class StableDiffusionMultiControlNetPipelineFastTests(
295291
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
296292
image_params = frozenset([]) # TO_DO: add image_params once refactored VaeImageProcessor.preprocess
297293

294+
supports_dduf = False
295+
298296
def get_dummy_components(self):
299297
torch.manual_seed(0)
300298
unet = UNet2DConditionModel(
@@ -518,10 +516,6 @@ def test_inference_multiple_prompt_input(self):
518516

519517
assert image.shape == (4, 64, 64, 3)
520518

521-
@unittest.skip("Test not supported.")
522-
def test_save_load_dduf(self):
523-
pass
524-
525519

526520
class StableDiffusionMultiControlNetOneModelPipelineFastTests(
527521
IPAdapterTesterMixin, PipelineTesterMixin, PipelineKarrasSchedulerTesterMixin, unittest.TestCase
@@ -531,6 +525,8 @@ class StableDiffusionMultiControlNetOneModelPipelineFastTests(
531525
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
532526
image_params = frozenset([]) # TO_DO: add image_params once refactored VaeImageProcessor.preprocess
533527

528+
supports_dduf = False
529+
534530
def get_dummy_components(self):
535531
torch.manual_seed(0)
536532
unet = UNet2DConditionModel(
@@ -705,10 +701,6 @@ def test_save_pretrained_raise_not_implemented_exception(self):
705701
except NotImplementedError:
706702
pass
707703

708-
@unittest.skip("Test not supported.")
709-
def test_save_load_dduf(self):
710-
pass
711-
712704

713705
@slow
714706
@require_torch_gpu

‎tests/pipelines/controlnet/test_controlnet_blip_diffusion.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,8 @@ class BlipDiffusionControlNetPipelineFastTests(PipelineTesterMixin, unittest.Tes
6868
"prompt_reps",
6969
]
7070

71+
supports_dduf = False
72+
7173
def get_dummy_components(self):
7274
torch.manual_seed(0)
7375
text_encoder_config = CLIPTextConfig(
@@ -220,7 +222,3 @@ def test_blipdiffusion_controlnet(self):
220222
assert (
221223
np.abs(image_slice.flatten() - expected_slice).max() < 1e-2
222224
), f" expected_slice {expected_slice}, but got {image_slice.flatten()}"
223-
224-
@unittest.skip("Test not supported.")
225-
def test_save_load_dduf(self):
226-
pass

‎tests/pipelines/controlnet/test_controlnet_img2img.py‎

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -189,10 +189,6 @@ def test_xformers_attention_forwardGenerator_pass(self):
189189
def test_inference_batch_single_identical(self):
190190
self._test_inference_batch_single_identical(expected_max_diff=2e-3)
191191

192-
@unittest.skip("Test not supported.")
193-
def test_save_load_dduf(self):
194-
pass
195-
196192

197193
class StableDiffusionMultiControlNetPipelineFastTests(
198194
IPAdapterTesterMixin, PipelineTesterMixin, PipelineKarrasSchedulerTesterMixin, unittest.TestCase
@@ -393,10 +389,6 @@ def test_save_pretrained_raise_not_implemented_exception(self):
393389
except NotImplementedError:
394390
pass
395391

396-
@unittest.skip("Test not supported.")
397-
def test_save_load_dduf(self):
398-
pass
399-
400392

401393
@slow
402394
@require_torch_gpu

‎tests/pipelines/controlnet/test_controlnet_inpaint.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -257,6 +257,8 @@ class MultiControlNetInpaintPipelineFastTests(
257257
params = TEXT_GUIDED_IMAGE_INPAINTING_PARAMS
258258
batch_params = TEXT_GUIDED_IMAGE_INPAINTING_BATCH_PARAMS
259259

260+
supports_dduf = False
261+
260262
def get_dummy_components(self):
261263
torch.manual_seed(0)
262264
unet = UNet2DConditionModel(
@@ -441,10 +443,6 @@ def test_save_pretrained_raise_not_implemented_exception(self):
441443
except NotImplementedError:
442444
pass
443445

444-
@unittest.skip("Test not supported.")
445-
def test_save_load_dduf(self):
446-
pass
447-
448446

449447
@slow
450448
@require_torch_gpu

‎tests/pipelines/controlnet/test_controlnet_inpaint_sdxl.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,8 @@ class ControlNetPipelineSDXLFastTests(
7878
}
7979
)
8080

81+
supports_dduf = False
82+
8183
def get_dummy_components(self):
8284
torch.manual_seed(0)
8385
unet = UNet2DConditionModel(
@@ -352,7 +354,3 @@ def test_save_load_optional_components(self):
352354

353355
def test_float16_inference(self):
354356
super().test_float16_inference(expected_max_diff=5e-1)
355-
356-
@unittest.skip("Test not supported.")
357-
def test_save_load_dduf(self):
358-
pass

‎tests/pipelines/controlnet/test_controlnet_sdxl.py‎

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -478,10 +478,6 @@ def new_step(self, *args, **kwargs):
478478
]:
479479
assert_run_mixture(steps, split, scheduler_cls_timesteps[0], scheduler_cls_timesteps[1])
480480

481-
@unittest.skip("Test not supported.")
482-
def test_save_load_dduf(self):
483-
pass
484-
485481

486482
class StableDiffusionXLMultiControlNetPipelineFastTests(
487483
PipelineTesterMixin, PipelineKarrasSchedulerTesterMixin, SDXLOptionalComponentsTesterMixin, unittest.TestCase
@@ -491,6 +487,8 @@ class StableDiffusionXLMultiControlNetPipelineFastTests(
491487
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
492488
image_params = frozenset([]) # TO_DO: add image_params once refactored VaeImageProcessor.preprocess
493489

490+
supports_dduf = False
491+
494492
def get_dummy_components(self):
495493
torch.manual_seed(0)
496494
unet = UNet2DConditionModel(
@@ -687,10 +685,6 @@ def test_inference_batch_single_identical(self):
687685
def test_save_load_optional_components(self):
688686
return self._test_save_load_optional_components()
689687

690-
@unittest.skip("Test not supported.")
691-
def test_save_load_dduf(self):
692-
pass
693-
694688

695689
class StableDiffusionXLMultiControlNetOneModelPipelineFastTests(
696690
PipelineKarrasSchedulerTesterMixin, PipelineTesterMixin, SDXLOptionalComponentsTesterMixin, unittest.TestCase
@@ -700,6 +694,8 @@ class StableDiffusionXLMultiControlNetOneModelPipelineFastTests(
700694
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
701695
image_params = frozenset([]) # TO_DO: add image_params once refactored VaeImageProcessor.preprocess
702696

697+
supports_dduf = False
698+
703699
def get_dummy_components(self):
704700
torch.manual_seed(0)
705701
unet = UNet2DConditionModel(
@@ -895,10 +891,6 @@ def test_negative_conditions(self):
895891

896892
self.assertTrue(np.abs(image_slice_without_neg_cond - image_slice_with_neg_cond).max() > 1e-2)
897893

898-
@unittest.skip("Test not supported.")
899-
def test_save_load_dduf(self):
900-
pass
901-
902894

903895
@slow
904896
@require_torch_gpu

‎tests/pipelines/controlnet/test_controlnet_sdxl_img2img.py‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -361,7 +361,3 @@ def test_stable_diffusion_xl_prompt_embeds(self):
361361

362362
# make sure that it's equal
363363
assert np.abs(image_slice_1.flatten() - image_slice_2.flatten()).max() < 1e-4
364-
365-
@unittest.skip("Test not supported.")
366-
def test_save_load_dduf(self):
367-
pass

0 commit comments

Comments
 (0)