[diffusion] fix: fix ZImage SP sharding for 5D latents and unpad frames (#16418)

Signed-off-by: Chi <chixie.mcisaac@gmail.com>
2026-01-07 13:24:04 +08:00
parent 52de807dd7
commit f066036c8b
3 changed files with 52 additions and 0 deletions
@@ -578,6 +578,16 @@ TWO_GPU_CASES_B = [
        ),
        T2I_sampling_params,
    ),
+    DiffusionTestCase(
+        "zimage_image_t2i_2_gpus",
+        DiffusionServerArgs(
+            model_path="Tongyi-MAI/Z-Image-Turbo",
+            modality="image",
+            num_gpus=2,
+            ulysses_degree=2,
+        ),
+        T2I_sampling_params,
+    ),
    DiffusionTestCase(
        "flux_image_t2i_2_gpus",
        DiffusionServerArgs(