diff --git a/utils/fetch_torch_cuda_pipeline_test_matrix.py b/utils/fetch_torch_cuda_pipeline_test_matrix.py index 227a60bc596f..196f35628ac1 100644 --- a/utils/fetch_torch_cuda_pipeline_test_matrix.py +++ b/utils/fetch_torch_cuda_pipeline_test_matrix.py @@ -12,12 +12,14 @@ PATH_TO_REPO = Path(__file__).parent.parent.resolve() ALWAYS_TEST_PIPELINE_MODULES = [ "controlnet", + "controlnet_flux", + "controlnet_sd3", "stable_diffusion", "stable_diffusion_2", + "stable_diffusion_3", "stable_diffusion_xl", - "stable_diffusion_adapter", "ip_adapters", - "kandinsky2_2", + "flux", ] PIPELINE_USAGE_CUTOFF = int(os.getenv("PIPELINE_USAGE_CUTOFF", 50000))