Add UnCLIPImageVariationPipeline to dummy imports (huggingface#1897)

* Add UnCLIPImageVariationPipeline to dummy imports * style
0xdigiscore · Jan 3, 2023 · f17fae6 · f17fae6
1 parent da31075
commit f17fae6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/pipelines/unclip/__init__.py b/src/diffusers/pipelines/unclip/__init__.py
@@ -10,7 +10,7 @@
     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