AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

fix test skipping

Browse Source
This commit is contained in:
BuildTools
2025-06-13 05:35:58 -06:00
parent 00ebba9725
commit 2b6722ecea
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/pipelines/test_pipelines_common.py
View File

@@ -1208,8 +1208,9 @@ class PipelineTesterMixin:
f"Required optional parameters not present: {remaining_required_optional_parameters}",
)
@unittest.skipIf("Chroma" in pipeline_class.__name__, reason="Chroma does not support batching")
def test_inference_batch_consistent(self, batch_sizes=[2]):
if "Chroma" in self.pipeline_class.__name__:
self.skipTest("Chroma does not support batching")
self._test_inference_batch_consistent(batch_sizes=batch_sizes)
def _test_inference_batch_consistent(
@@ -1261,8 +1262,9 @@ class PipelineTesterMixin:
output = pipe(**batched_input)
assert len(output[0]) == batch_size
@unittest.skipIf("Chroma" in pipeline_class.__name__, reason="Chroma does not support batching")
def test_inference_batch_single_identical(self, batch_size=3, expected_max_diff=1e-4):
if "Chroma" in self.pipeline_class.__name__:
self.skipTest("Chroma does not support batching")
self._test_inference_batch_single_identical(batch_size=batch_size, expected_max_diff=expected_max_diff)
def _test_inference_batch_single_identical(
@@ -1831,8 +1833,9 @@ class PipelineTesterMixin:
if test_mean_pixel_difference:
assert_mean_pixel_difference(output_with_offload[0], output_without_offload[0])
@unittest.skipIf("Chroma" in pipeline_class.__name__, reason="Chroma does not yet support num_images_per_prompt")
def test_num_images_per_prompt(self):
if "Chroma" in self.pipeline_class.__name__:
self.skipTest("Chroma does not yet support num_images_per_prompt")
sig = inspect.signature(self.pipeline_class.__call__)
if "num_images_per_prompt" not in sig.parameters: