diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py index e4095f615e..4b85f2662a 100644 --- a/src/diffusers/__init__.py +++ b/src/diffusers/__init__.py @@ -223,6 +223,7 @@ else: "FluxControlNetModel", "FluxMultiControlNetModel", "FluxTransformer2DModel", + "GlmImageDecoderTransformer2DModel", "HiDreamImageTransformer2DModel", "HunyuanDiT2DControlNetModel", "HunyuanDiT2DModel", @@ -970,6 +971,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: FluxControlNetModel, FluxMultiControlNetModel, FluxTransformer2DModel, + GlmImageDecoderTransformer2DModel, HiDreamImageTransformer2DModel, HunyuanDiT2DControlNetModel, HunyuanDiT2DModel,