diff --git a/src/diffusers/pipelines/stable_diffusion/__init__.py b/src/diffusers/pipelines/stable_diffusion/__init__.py index 91cdab0a162e..3c012dbab89d 100644 --- a/src/diffusers/pipelines/stable_diffusion/__init__.py +++ b/src/diffusers/pipelines/stable_diffusion/__init__.py @@ -42,7 +42,7 @@ class StableDiffusionPipelineOutput(BaseOutput): from .pipeline_stable_diffusion_inpaint_legacy import StableDiffusionInpaintPipelineLegacy from .safety_checker import StableDiffusionSafetyChecker -if is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0."): +if is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0.dev0"): from .pipeline_stable_diffusion_image_variation import StableDiffusionImageVariationPipeline else: from ...utils.dummy_torch_and_transformers_objects import StableDiffusionImageVariationPipeline diff --git a/src/diffusers/pipelines/versatile_diffusion/__init__.py b/src/diffusers/pipelines/versatile_diffusion/__init__.py index 7865c6283494..1d2caa7e2399 100644 --- a/src/diffusers/pipelines/versatile_diffusion/__init__.py +++ b/src/diffusers/pipelines/versatile_diffusion/__init__.py @@ -1,7 +1,7 @@ from ...utils import is_torch_available, is_transformers_available, is_transformers_version -if is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0."): +if is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0.dev0"): from .modeling_text_unet import UNetFlatConditionModel from .pipeline_versatile_diffusion import VersatileDiffusionPipeline from .pipeline_versatile_diffusion_dual_guided import VersatileDiffusionDualGuidedPipeline diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index ad1e8a900253..c0294b4a3d23 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -308,7 +308,7 @@ def requires_backends(obj, backends): "VersatileDiffusionPipeline", "VersatileDiffusionDualGuidedPipeline", "StableDiffusionImageVariationPipeline", - ] and is_transformers_version("<", "4.25.0."): + ] and is_transformers_version("<", "4.25.0.dev0"): raise ImportError( f"You need to install `transformers` from 'main' in order to use {name}: \n```\n pip install" " git+https://github.com/huggingface/transformers \n```"