From 152f7ca357c066c4af3d1a58cdf17662ef5a2f87 Mon Sep 17 00:00:00 2001
From: DefTruth <31974251+DefTruth@users.noreply.github.com>
Date: Sat, 29 Nov 2025 17:28:33 +0800
Subject: [PATCH] fix type-check for z-image transformer (#12739)

* allow type-check for ZImageTransformer2DModel

* make fix-copies
---
 src/diffusers/__init__.py               |  1 +
 src/diffusers/utils/dummy_pt_objects.py | 15 +++++++++++++++
 2 files changed, 16 insertions(+)

diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py
index 8a81beca97..f2d1840da2 100644
--- a/src/diffusers/__init__.py
+++ b/src/diffusers/__init__.py
@@ -991,6 +991,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
             WanAnimateTransformer3DModel,
             WanTransformer3DModel,
             WanVACETransformer3DModel,
+            ZImageTransformer2DModel,
             attention_backend,
         )
         from .modular_pipelines import ComponentsManager, ComponentSpec, ModularPipeline, ModularPipelineBlocks
diff --git a/src/diffusers/utils/dummy_pt_objects.py b/src/diffusers/utils/dummy_pt_objects.py
index 928f0b9774..73854b3819 100644
--- a/src/diffusers/utils/dummy_pt_objects.py
+++ b/src/diffusers/utils/dummy_pt_objects.py
@@ -1698,6 +1698,21 @@ class WanVACETransformer3DModel(metaclass=DummyObject):
         requires_backends(cls, ["torch"])
 
 
+class ZImageTransformer2DModel(metaclass=DummyObject):
+    _backends = ["torch"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch"])
+
+
 def attention_backend(*args, **kwargs):
     requires_backends(attention_backend, ["torch"])