[Tests] Run diffusion tiny model tests in parallel (#6339)

Signed-off-by: Nick Cao <ncao@redhat.com>
This commit is contained in:
Nick Cao authored and GitHub committed 2026-08-20 12:46:32 -04:00
1 parent c6e38860e9
commit ebb4abde0a
7 files changed
+68 -28

No files matched your search

@@ -24,7 +24,7 @@ from tests.model_tests.diffusion.task_runners import (
)
# NOTE : Hardware marks are added dynamically based on test requirements
pytestmark = [pytest.mark.diffusion]
pytestmark = [pytest.mark.diffusion, pytest.mark.xdist]
@pytest.mark.parametrize(
@@ -58,7 +58,7 @@ def test_online_on_supported_tasks(
log_stats=server.log_stats,
)
for task_type in supported_tasks:
with subtests.test(msg=task_type):
with subtests.test(msg=task_type.value):
if task_type == DiffusionTasks.TEXT_TO_IMAGE:
run_and_validate_online_text_to_image_request(server, client)
elif task_type == DiffusionTasks.IMAGE_TO_IMAGE: