[diffusion] feat: support passing component path via server args (#19108)

This commit is contained in:
Mick
2026-02-21 21:22:47 +08:00
committed by GitHub
parent c36a10aabb
commit 6503f94211
18 changed files with 170 additions and 77 deletions
@@ -226,6 +226,7 @@ class DiffusionTestCase:
id: str # pytest test id and scenario name
server_args: DiffusionServerArgs
sampling_params: DiffusionSamplingParams
run_perf_check: bool = True
def sample_step_indices(
@@ -507,6 +508,16 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
),
TI2I_sampling_params,
),
DiffusionTestCase(
"flux_2_t2i_customized_vae_path",
DiffusionServerArgs(
model_path="black-forest-labs/FLUX.2-dev",
modality="image",
extras=["--vae-path=fal/FLUX.2-Tiny-AutoEncoder"],
),
T2I_sampling_params,
run_perf_check=False,
),
DiffusionTestCase(
"fast_hunyuan_video",
DiffusionServerArgs(