From 7d7e18b9cced4acecfd2981bbfb446a45925bbb3 Mon Sep 17 00:00:00 2001 From: DN6 Date: Fri, 21 Feb 2025 13:17:33 +0530 Subject: [PATCH 1/2] update --- utils/fetch_torch_cuda_pipeline_test_matrix.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/utils/fetch_torch_cuda_pipeline_test_matrix.py b/utils/fetch_torch_cuda_pipeline_test_matrix.py index 227a60bc596f..f7e813673c2f 100644 --- a/utils/fetch_torch_cuda_pipeline_test_matrix.py +++ b/utils/fetch_torch_cuda_pipeline_test_matrix.py @@ -14,10 +14,12 @@ "controlnet", "stable_diffusion", "stable_diffusion_2", + "stable_diffusion_3", "stable_diffusion_xl", - "stable_diffusion_adapter", "ip_adapters", - "kandinsky2_2", + "flux", + "hunyuan_dit", + "hunyuan_video", ] PIPELINE_USAGE_CUTOFF = int(os.getenv("PIPELINE_USAGE_CUTOFF", 50000)) From 9e56d656df8270eb3e6ff5c9d63cecbc5c708b34 Mon Sep 17 00:00:00 2001 From: DN6 Date: Fri, 21 Feb 2025 14:00:02 +0530 Subject: [PATCH 2/2] update --- utils/fetch_torch_cuda_pipeline_test_matrix.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/fetch_torch_cuda_pipeline_test_matrix.py b/utils/fetch_torch_cuda_pipeline_test_matrix.py index f7e813673c2f..196f35628ac1 100644 --- a/utils/fetch_torch_cuda_pipeline_test_matrix.py +++ b/utils/fetch_torch_cuda_pipeline_test_matrix.py @@ -12,14 +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", "ip_adapters", "flux", - "hunyuan_dit", - "hunyuan_video", ] PIPELINE_USAGE_CUTOFF = int(os.getenv("PIPELINE_USAGE_CUTOFF", 50000))