Skip to content

Commit 9ef1185

Browse files
authored
[tests] disable xformer tests for pipelines it isn't popular. (huggingface#12277)
disable xformer tests for pipelines it isn't popular.
1 parent 7c54a7b commit 9ef1185

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

tests/pipelines/easyanimate/test_easyanimate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ class EasyAnimatePipelineFastTests(PipelineTesterMixin, unittest.TestCase):
4848
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
4949
image_params = TEXT_TO_IMAGE_IMAGE_PARAMS
5050
image_latents_params = TEXT_TO_IMAGE_IMAGE_PARAMS
51+
test_xformers_attention = False
5152
required_optional_params = frozenset(
5253
[
5354
"num_inference_steps",

tests/pipelines/hidream_image/test_pipeline_hidream.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,8 @@ class HiDreamImagePipelineFastTests(PipelineTesterMixin, unittest.TestCase):
4747
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
4848
image_params = TEXT_TO_IMAGE_IMAGE_PARAMS
4949
image_latents_params = TEXT_TO_IMAGE_IMAGE_PARAMS
50-
5150
required_optional_params = PipelineTesterMixin.required_optional_params
51+
test_xformers_attention = False
5252
test_layerwise_casting = True
5353
supports_dduf = False
5454

tests/pipelines/omnigen/test_pipeline_omnigen.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ class OmniGenPipelineFastTests(unittest.TestCase, PipelineTesterMixin):
2222
pipeline_class = OmniGenPipeline
2323
params = frozenset(["prompt", "guidance_scale"])
2424
batch_params = frozenset(["prompt"])
25-
25+
test_xformers_attention = False
2626
test_layerwise_casting = True
2727

2828
def get_dummy_components(self):

tests/pipelines/qwenimage/test_qwenimage_controlnet.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,6 @@ class QwenControlNetPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
4444
batch_params = frozenset(["prompt", "negative_prompt", "control_image"])
4545
image_params = frozenset(["control_image"])
4646
image_latents_params = frozenset(["latents"])
47-
4847
required_optional_params = frozenset(
4948
[
5049
"num_inference_steps",
@@ -59,7 +58,7 @@ class QwenControlNetPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
5958
)
6059

6160
supports_dduf = False
62-
test_xformers_attention = True
61+
test_xformers_attention = False
6362
test_layerwise_casting = True
6463
test_group_offloading = True
6564

0 commit comments

Comments
 (0)