From 09de36c4d881cff52718bd94ba3a4dcb4e0caf37 Mon Sep 17 00:00:00 2001
From: DN6 <dhruv.nair@gmail.com>
Date: Mon, 21 Apr 2025 15:33:50 +0530
Subject: [PATCH] update

---
 src/diffusers/pipelines/deprecated/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/diffusers/pipelines/deprecated/__init__.py b/src/diffusers/pipelines/deprecated/__init__.py
index 569108f873..04c76a15e8 100644
--- a/src/diffusers/pipelines/deprecated/__init__.py
+++ b/src/diffusers/pipelines/deprecated/__init__.py
@@ -124,6 +124,7 @@ else:
         "UniDiffuserModel",
         "UTransformer2DModel",
         "UniDiffuserPipeline",
+        "ImageTextPipelineOutput",
     ]
     _import_structure["versatile_diffusion"] = [
         "VersatileDiffusionDualGuidedPipeline",
@@ -221,7 +222,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
             VideoToVideoSDPipeline,
         )
         from .unclip import UnCLIPImageVariationPipeline, UnCLIPPipeline
-        from .unidiffuser import UniDiffuserPipeline
+        from .unidiffuser import ImageTextPipelineOutput, UniDiffuserPipeline
         from .versatile_diffusion import (
             VersatileDiffusionDualGuidedPipeline,
             VersatileDiffusionImageVariationPipeline,