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], + ], ), }