[diffusion] CI: fix CI script path and missing server arg in perf baseline generator (#20138)
This commit is contained in:
4
.github/workflows/diffusion-ci-gt-gen.yml
vendored
4
.github/workflows/diffusion-ci-gt-gen.yml
vendored
@@ -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: |
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user