[diffusion] fix: fix corrupted image editing outputs in Multi-GPU SP mode for FLUX.2-klein models (#19454)

This commit is contained in:
Ruihang Li
2026-03-04 11:35:46 +08:00
committed by GitHub
parent 6851613b93
commit da9dcbc906
7 changed files with 128 additions and 24 deletions
@@ -790,6 +790,15 @@ TWO_GPU_CASES_B = [
),
T2I_sampling_params,
),
DiffusionTestCase(
"flux_2_klein_ti2i_2_gpus",
DiffusionServerArgs(
model_path="black-forest-labs/FLUX.2-klein-4B",
modality="image",
num_gpus=2,
),
TI2I_sampling_params,
),
]
# Load global configuration