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: