From 6fb2b36a8165bc3dc320421a43166bf117663ae9 Mon Sep 17 00:00:00 2001 From: Khairul Kabir <250204754+khairulkabir1661@users.noreply.github.com> Date: Thu, 24 Sep 2026 16:19:45 -0700 Subject: [PATCH] [CI][Diffusion] Add acceleration test groups for SD3 and Flux2 (#6625) Signed-off-by: khairulkabir1661 Co-authored-by: khairulkabir1661 Co-authored-by: Claude Opus 4.6 --- tests/model_tests/diffusion/model_settings.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/model_tests/diffusion/model_settings.py b/tests/model_tests/diffusion/model_settings.py index 85ec720be..fb5376046 100644 --- a/tests/model_tests/diffusion/model_settings.py +++ b/tests/model_tests/diffusion/model_settings.py @@ -135,6 +135,11 @@ DIFFUSION_TEST_SETTINGS = { model="black-forest-labs/FLUX.2-dev", builder=diff_model_builders.tiny_flux2_builder, supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE], + extra_test_groups=[ + [DiffusionAccs.CACHE_DIT, DiffusionAccs.LAYERWISE_OFFLOAD], + [DiffusionAccs.CFG_PARALLEL, DiffusionAccs.SEQUENCE_PARALLEL, DiffusionAccs.CPU_OFFLOAD], + [DiffusionAccs.TENSOR_PARALLEL, DiffusionAccs.VAE_PATCH_PARALLEL], + ], ), "QwenImageEditPipeline": DiffusionModelTestOpts( model="Qwen/Qwen-Image-Edit", @@ -150,5 +155,9 @@ DIFFUSION_TEST_SETTINGS = { model="stabilityai/stable-diffusion-3.5-medium", builder=diff_model_builders.tiny_sd3_builder, supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE], + extra_test_groups=[ + [DiffusionAccs.CACHE_DIT, DiffusionAccs.LAYERWISE_OFFLOAD], + [DiffusionAccs.CFG_PARALLEL, DiffusionAccs.TENSOR_PARALLEL, DiffusionAccs.CPU_OFFLOAD], + ], ), }