Add PhotonTransformer2DModel to TYPE_CHECKING imports

DavidBert · DavidBert · commit 7fefe099e15c · 2025-10-20T07:27:03.000Z
diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py
@@ -923,6 +923,7 @@
             MultiControlNetModel,
             OmniGenTransformer2DModel,
             ParallelConfig,
+            PhotonTransformer2DModel,
             PixArtTransformer2DModel,
             PriorTransformer,
             QwenImageControlNetModel,
