Add PhotonTransformer2DModel to TYPE_CHECKING imports

2026-01-27 17:22:53 +03:00 · 2025-10-20 07:27:03 +00:00
parent d5ffd35d70
commit fdc8e34533
1 changed files with 1 additions and 0 deletions
--- a/src/diffusers/init.py
+++ b/src/diffusers/init.py
@@ -928,6 +928,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
            MultiControlNetModel,
            OmniGenTransformer2DModel,
            ParallelConfig,
+            PhotonTransformer2DModel,
            PixArtTransformer2DModel,
            PriorTransformer,
            QwenImageControlNetModel,