[Tests] Config driven tiny model builder (#5090)

Signed-off-by: Nick Cao <ncao@redhat.com>
This commit is contained in:
Nick Cao authored and GitHub committed 2026-07-22 17:11:53 +00:00
1 parent cc1939120f
commit f10320f53f
18 files changed
+601 -60

No files matched your search

@@ -12,6 +12,7 @@ from tests.model_tests.diffusion.task_runners import (
run_and_validate_text_to_image_determinism,
run_and_validate_text_to_image_multi_output,
run_and_validate_text_to_image_request,
run_and_validate_text_to_video_request,
)
# NOTE : Hardware marks are added dynamically based on test requirements
@@ -53,6 +54,8 @@ def test_pipeline_on_supported_tasks(
run_and_validate_text_to_image_request(omni)
elif task_type == DiffusionTasks.IMAGE_TO_IMAGE:
run_and_validate_image_to_image_request(omni)
elif task_type == DiffusionTasks.TEXT_TO_VIDEO:
run_and_validate_text_to_video_request(omni)
else:
raise ValueError(f"Task type {task_type} is not yet supported")