ci: enable /rerun-test for multimodal gen PR tests (#22828)

.github/workflows/rerun-test.yml

@@ -38,6 +38,11 @@ on:
         required: false
         type: string
         default: "false"
+      install_diffusion:
+        description: "Install diffusion dependencies (for multimodal gen tests)"
+        required: false
+        type: string
+        default: "false"
 
 env:
   SGLANG_IS_IN_CI: true
@@ -73,6 +78,8 @@ jobs:
           fi
           if [[ "${{ inputs.use_deepep }}" == "true" ]]; then
             bash scripts/ci/cuda/ci_install_deepep.sh
+          elif [[ "${{ inputs.install_diffusion }}" == "true" ]]; then
+            bash scripts/ci/cuda/ci_install_dependency.sh diffusion
           else
             bash scripts/ci/cuda/ci_install_dependency.sh
           fi
@@ -83,7 +90,6 @@ jobs:
           if [[ "${{ inputs.runner_label }}" == "1-gpu-5090" ]]; then
             source /etc/profile.d/sglang-ci.sh
           fi
-          cd test/
           # Collect non-empty commands into an array for counting.
           cmds=()
           while IFS= read -r cmd; do
@@ -97,10 +103,17 @@ jobs:
             cmd="${cmds[$idx]}"
             echo ""
             echo "."
-            echo "Begin ($i/$total): python3 $cmd"
-            echo "."
-            file_start=$SECONDS
-            python3 $cmd -f || exit 1
+            if [[ "${{ inputs.install_diffusion }}" == "true" ]]; then
+              echo "Begin ($i/$total): python3 -m pytest $cmd -x"
+              echo "."
+              file_start=$SECONDS
+              python3 -m pytest $cmd -x || exit 1
+            else
+              echo "Begin ($i/$total): python3 $cmd"
+              echo "."
+              file_start=$SECONDS
+              (cd test/ && python3 $cmd -f) || exit 1
+            fi
             elapsed=$(( SECONDS - file_start ))
             echo "."
             echo "End ($i/$total): elapsed=${elapsed}s"