From fdc8e345333db51f793e339bdc1bdc2360fe9c41 Mon Sep 17 00:00:00 2001
From: David Bertoin <davidbertoin@gmail.com>
Date: Mon, 20 Oct 2025 07:27:03 +0000
Subject: [PATCH] Add PhotonTransformer2DModel to TYPE_CHECKING imports

---
 src/diffusers/__init__.py | 1 +
 1 file changed, 1 insertion(+)

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,