[diffusion] feat: support passing component path via server args (#19108)
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user