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

skip batch tests

Browse Source
This commit is contained in:
BuildTools
2025-06-13 05:25:11 -06:00
parent bea8b0d86e
commit 00ebba9725
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/pipelines/test_pipelines_common.py
View File

@@ -1208,6 +1208,7 @@ 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]):
self._test_inference_batch_consistent(batch_sizes=batch_sizes)
@@ -1260,6 +1261,7 @@ 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):
self._test_inference_batch_single_identical(batch_size=batch_size, expected_max_diff=expected_max_diff)
@@ -1829,6 +1831,7 @@ 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):
sig = inspect.signature(self.pipeline_class.__call__)