[diffusion] fix: fix corrupted image editing outputs in Multi-GPU SP mode for FLUX.2-klein models (#19454)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user