[CI] fix: benchmark cannot run as expected

Author: alphabetc1

.gitignore

@@ -188,6 +188,9 @@ cython_debug/
 #  you could uncomment the following to ignore the entire vscode folder
 .vscode/
 
+# sglang diffusion worker generated outputs
+outputs/
+
 # Ruff stuff:
 .ruff_cache/
 

development.md

@@ -9,4 +9,4 @@ Run tests:
 ```bash
 pip install pytest
 pytest tests/unit -v
-```
+```

outputs/7700f966-bca5-406d-a6ad-bdfaf283eedb.png

@@ -4,10 +4,10 @@
 
 Example:
   python tests/benchmarks/diffusion_router/bench_router.py \
-    --model Wan-AI/Wan2.2-T2V-A14B-Diffusers \
+    --model Qwen/Qwen-Image \
     --num-workers 2 \
     --num-prompts 20 \
-    --max-concurrency 4
+    --max-concurrency 2
 """
 
 from __future__ import annotations
@@ -201,19 +201,15 @@ def _build_bench_command(args: argparse.Namespace, base_url: str) -> list[str]:
     cmd = [
         sys.executable,
         "-m",
-        "sglang.bench_serving",
-        "--backend",
-        "sglang",
+        "sglang.multimodal_gen.benchmarks.bench_serving",
         "--base-url",
         base_url,
         "--model",
         args.model,
-        "--dataset-name",
+        "--dataset",
         args.dataset,
         "--num-prompts",
         str(args.num_prompts),
-        "--max-concurrency",
-        str(args.max_concurrency),
         "--request-rate",
         str(args.request_rate),
         "--log-level",
@@ -222,6 +218,8 @@ def _build_bench_command(args: argparse.Namespace, base_url: str) -> list[str]:
 
     if args.dataset_path:
         cmd += ["--dataset-path", args.dataset_path]
+    if args.max_concurrency:
+        cmd += ["--max-concurrency", str(args.max_concurrency)]
     if args.task:
         cmd += ["--task", args.task]
     if args.width:

outputs/a6158412-05b0-4d73-a5b6-c2ce3af29718.jpg

@@ -5,7 +5,7 @@
 
 Example:
   python tests/benchmarks/diffusion_router/bench_routing_algorithms.py \
-    --model Wan-AI/Wan2.2-T2V-A14B-Diffusers \
+    --model Qwen/Qwen-Image \
     --num-workers 2 \
     --num-prompts 10 \
     --max-concurrency 2