From f17fae641c7c268e60c479396891ea97a6b29aae Mon Sep 17 00:00:00 2001 From: Anton Lozhkov Date: Tue, 3 Jan 2023 11:57:56 +0100 Subject: [PATCH] Add UnCLIPImageVariationPipeline to dummy imports (#1897) * Add UnCLIPImageVariationPipeline to dummy imports * style --- src/diffusers/pipelines/unclip/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/unclip/__init__.py b/src/diffusers/pipelines/unclip/__init__.py index 23b54a7d2f..075e66bb68 100644 --- a/src/diffusers/pipelines/unclip/__init__.py +++ b/src/diffusers/pipelines/unclip/__init__.py @@ -10,7 +10,7 @@ try: if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")): raise OptionalDependencyNotAvailable() except OptionalDependencyNotAvailable: - from ...utils.dummy_torch_and_transformers_objects import UnCLIPPipeline + from ...utils.dummy_torch_and_transformers_objects import UnCLIPImageVariationPipeline, UnCLIPPipeline else: from .pipeline_unclip import UnCLIPPipeline from .pipeline_unclip_image_variation import UnCLIPImageVariationPipeline