[diffusion] CI: add testcase for cfg parallel (#17056)

This commit is contained in:
Mick
2026-01-15 13:13:31 +08:00
committed by GitHub
parent 000ad42225
commit 68e8d0f68d
2 changed files with 83 additions and 1 deletions
@@ -149,6 +149,7 @@ class DiffusionServerArgs:
tp_size: int | None = None
ulysses_degree: int | None = None
ring_degree: int | None = None
cfg_parallel: bool | None = None
# LoRA
lora_path: str | None = (
None # LoRA adapter path (HF repo or local path, loaded at startup)
@@ -576,6 +577,20 @@ TWO_GPU_CASES_A = [
output_size="832x480",
),
),
DiffusionTestCase(
"wan2_1_t2v_1.3b_cfg_parallel",
DiffusionServerArgs(
model_path="Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
modality="video",
warmup=0,
custom_validator="video",
num_gpus=2,
cfg_parallel=True,
),
DiffusionSamplingParams(
prompt=T2V_PROMPT,
),
),
]
# Skip turbowan because Triton requires 81920 shared memory, but AMD only has 65536.