Merged into one transformer for Z-Image.

Author: JerryWu-code

src/diffusers/__init__.py

@@ -279,7 +279,6 @@
             "WanTransformer3DModel",
             "WanVACETransformer3DModel",
             "ZImageTransformer2DModel",
-            "ZImageOmniTransformer2DModel",
             "attention_backend",
         ]
     )
@@ -1015,7 +1014,6 @@
             WanTransformer3DModel,
             WanVACETransformer3DModel,
             ZImageTransformer2DModel,
-            ZImageOmniTransformer2DModel,
             attention_backend,
         )
         from .modular_pipelines import ComponentsManager, ComponentSpec, ModularPipeline, ModularPipelineBlocks

src/diffusers/models/__init__.py

@@ -116,7 +116,6 @@
     _import_structure["transformers.transformer_wan_animate"] = ["WanAnimateTransformer3DModel"]
     _import_structure["transformers.transformer_wan_vace"] = ["WanVACETransformer3DModel"]
     _import_structure["transformers.transformer_z_image"] = ["ZImageTransformer2DModel"]
-    _import_structure["transformers.transformer_z_image_omni"] = ["ZImageOmniTransformer2DModel"]
     _import_structure["unets.unet_1d"] = ["UNet1DModel"]
     _import_structure["unets.unet_2d"] = ["UNet2DModel"]
     _import_structure["unets.unet_2d_condition"] = ["UNet2DConditionModel"]

src/diffusers/models/transformers/__init__.py

@@ -48,4 +48,3 @@
     from .transformer_wan_animate import WanAnimateTransformer3DModel
     from .transformer_wan_vace import WanVACETransformer3DModel
     from .transformer_z_image import ZImageTransformer2DModel
-    from .transformer_z_image_omni import ZImageOmniTransformer2DModel