From 547a017a77c94ab280b6dcbab73047863fee93d0 Mon Sep 17 00:00:00 2001
From: Mick <mickjagger19@icloud.com>
Date: Fri, 19 Jun 2026 13:43:02 +0800
Subject: [PATCH] [diffusion] CI: run nightly image comparisons on 2 GPUs
 (#28593)

---
 .../utils/diffusion/comparison_configs.json   | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/scripts/ci/utils/diffusion/comparison_configs.json b/scripts/ci/utils/diffusion/comparison_configs.json
index c8c4b2561..b123dc80b 100644
--- a/scripts/ci/utils/diffusion/comparison_configs.json
+++ b/scripts/ci/utils/diffusion/comparison_configs.json
@@ -10,10 +10,10 @@
             "width": 1024,
             "height": 1024,
             "seed": 42,
-            "num_gpus": 1,
+            "num_gpus": 2,
             "frameworks": {
                 "sglang": {
-                    "serve_args": "--warmup --dit-layerwise-offload false",
+                    "serve_args": "--warmup --dit-layerwise-offload false --tp-size 2",
                     "extra_env": {}
                 }
             }
@@ -26,10 +26,10 @@
             "width": 1024,
             "height": 1024,
             "seed": 42,
-            "num_gpus": 1,
+            "num_gpus": 2,
             "frameworks": {
                 "sglang": {
-                    "serve_args": "--warmup --dit-layerwise-offload false",
+                    "serve_args": "--warmup --dit-layerwise-offload false --tp-size 2",
                     "extra_env": {}
                 }
             }
@@ -42,10 +42,10 @@
             "width": 1024,
             "height": 1024,
             "seed": 42,
-            "num_gpus": 1,
+            "num_gpus": 2,
             "frameworks": {
                 "sglang": {
-                    "serve_args": "--warmup",
+                    "serve_args": "--warmup --tp-size 2",
                     "extra_env": {}
                 }
             }
@@ -59,10 +59,10 @@
             "width": 1024,
             "height": 1024,
             "seed": 42,
-            "num_gpus": 1,
+            "num_gpus": 2,
             "frameworks": {
                 "sglang": {
-                    "serve_args": "--warmup",
+                    "serve_args": "--warmup --tp-size 2",
                     "extra_env": {}
                 }
             }
@@ -75,10 +75,10 @@
             "width": 1024,
             "height": 1024,
             "seed": 42,
-            "num_gpus": 1,
+            "num_gpus": 2,
             "frameworks": {
                 "sglang": {
-                    "serve_args": "--warmup",
+                    "serve_args": "--warmup --tp-size 2",
                     "extra_env": {}
                 }
             }