diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py
index 28b2ae2549..b7086d2e0c 100644
--- 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,