Make diffusers importable with transformers < 4.26 (#2380)

a60f5555 · Pedro Cuenca · GitHub · 90a624f6 · a60f5555
Unverified Commit a60f5555 authored Feb 16, 2023 by Pedro Cuenca Committed by GitHub Feb 16, 2023
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

src/diffusers/pipelines/stable_diffusion/__init__.py src/diffusers/pipelines/stable_diffusion/__init__.py +4 -1

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/__init__.py
+++ b/src/diffusers/pipelines/stable_diffusion/__init__.py
@@ -64,7 +64,10 @@ try:
    if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.26.0")):
        raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
-    from ...utils.dummy_torch_and_transformers_objects import StableDiffusionDepth2ImgPipeline
+    from ...utils.dummy_torch_and_transformers_objects import (
+        StableDiffusionDepth2ImgPipeline,
+        StableDiffusionPix2PixZeroPipeline,
+    )
 else:
    from .pipeline_stable_diffusion_depth2img import StableDiffusionDepth2ImgPipeline
    from .pipeline_stable_diffusion_pix2pix_zero import StableDiffusionPix2PixZeroPipeline