"graphbolt/src/cuda/sampling_utils.hip" did not exist on "557a8f817a285acc9dfee91c19a19ff8728a8d76"
Unverified Commit 9ef11850 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[tests] disable xformer tests for pipelines it isn't popular. (#12277)

disable xformer tests for pipelines it isn't popular.
parent 7c54a7b3
...@@ -48,6 +48,7 @@ class EasyAnimatePipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -48,6 +48,7 @@ class EasyAnimatePipelineFastTests(PipelineTesterMixin, unittest.TestCase):
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
image_params = TEXT_TO_IMAGE_IMAGE_PARAMS image_params = TEXT_TO_IMAGE_IMAGE_PARAMS
image_latents_params = TEXT_TO_IMAGE_IMAGE_PARAMS image_latents_params = TEXT_TO_IMAGE_IMAGE_PARAMS
test_xformers_attention = False
required_optional_params = frozenset( required_optional_params = frozenset(
[ [
"num_inference_steps", "num_inference_steps",
......
...@@ -47,8 +47,8 @@ class HiDreamImagePipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -47,8 +47,8 @@ class HiDreamImagePipelineFastTests(PipelineTesterMixin, unittest.TestCase):
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS batch_params = TEXT_TO_IMAGE_BATCH_PARAMS
image_params = TEXT_TO_IMAGE_IMAGE_PARAMS image_params = TEXT_TO_IMAGE_IMAGE_PARAMS
image_latents_params = TEXT_TO_IMAGE_IMAGE_PARAMS image_latents_params = TEXT_TO_IMAGE_IMAGE_PARAMS
required_optional_params = PipelineTesterMixin.required_optional_params required_optional_params = PipelineTesterMixin.required_optional_params
test_xformers_attention = False
test_layerwise_casting = True test_layerwise_casting = True
supports_dduf = False supports_dduf = False
......
...@@ -22,7 +22,7 @@ class OmniGenPipelineFastTests(unittest.TestCase, PipelineTesterMixin): ...@@ -22,7 +22,7 @@ class OmniGenPipelineFastTests(unittest.TestCase, PipelineTesterMixin):
pipeline_class = OmniGenPipeline pipeline_class = OmniGenPipeline
params = frozenset(["prompt", "guidance_scale"]) params = frozenset(["prompt", "guidance_scale"])
batch_params = frozenset(["prompt"]) batch_params = frozenset(["prompt"])
test_xformers_attention = False
test_layerwise_casting = True test_layerwise_casting = True
def get_dummy_components(self): def get_dummy_components(self):
......
...@@ -44,7 +44,6 @@ class QwenControlNetPipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -44,7 +44,6 @@ class QwenControlNetPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
batch_params = frozenset(["prompt", "negative_prompt", "control_image"]) batch_params = frozenset(["prompt", "negative_prompt", "control_image"])
image_params = frozenset(["control_image"]) image_params = frozenset(["control_image"])
image_latents_params = frozenset(["latents"]) image_latents_params = frozenset(["latents"])
required_optional_params = frozenset( required_optional_params = frozenset(
[ [
"num_inference_steps", "num_inference_steps",
...@@ -59,7 +58,7 @@ class QwenControlNetPipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -59,7 +58,7 @@ class QwenControlNetPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
) )
supports_dduf = False supports_dduf = False
test_xformers_attention = True test_xformers_attention = False
test_layerwise_casting = True test_layerwise_casting = True
test_group_offloading = True test_group_offloading = True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment