[diffusion] CI: add testcase for cfg parallel (#17056)
This commit is contained in:
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user