From 1cb86f51712f36ad6cae291bb4f1f2b8b8a21806 Mon Sep 17 00:00:00 2001 From: Yuhao Yang <47235274+yhyang201@users.noreply.github.com> Date: Mon, 9 Mar 2026 10:35:21 +0800 Subject: [PATCH] [diffusion] CI: fix CI script path and missing server arg in perf baseline generator (#20138) --- .github/workflows/diffusion-ci-gt-gen.yml | 4 ++-- .../sglang/multimodal_gen/test/scripts/gen_perf_baselines.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/.github/workflows/diffusion-ci-gt-gen.yml b/.github/workflows/diffusion-ci-gt-gen.yml index ef039180b..b1af552f9 100644 --- a/.github/workflows/diffusion-ci-gt-gen.yml +++ b/.github/workflows/diffusion-ci-gt-gen.yml @@ -37,7 +37,7 @@ jobs: ref: ${{ inputs.ref || github.ref }} - name: Install dependencies - run: bash scripts/ci/ci_install_dependency.sh diffusion + run: bash scripts/ci/cuda/ci_install_dependency.sh diffusion - name: Generate outputs run: | @@ -71,7 +71,7 @@ jobs: ref: ${{ inputs.ref || github.ref }} - name: Install dependencies - run: bash scripts/ci/ci_install_dependency.sh diffusion + run: bash scripts/ci/cuda/ci_install_dependency.sh diffusion - name: Generate outputs run: | diff --git a/python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py b/python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py index 982e51374..4d68124b9 100644 --- a/python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py +++ b/python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py @@ -61,6 +61,8 @@ def _build_server_extra_args(case: DiffusionTestCase) -> str: a += " --dit-layerwise-offload true" if server_args.dit_offload_prefetch_size: a += f" --dit-offload-prefetch-size {server_args.dit_offload_prefetch_size}" + if server_args.text_encoder_cpu_offload: + a += " --text-encoder-cpu-offload" if server_args.ring_degree is not None: a += f" --ring-degree {server_args.ring_degree}" if server_args.lora_path: