From 008c22d334ef978bc51df60de8a66b8613a81035 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 25 Jan 2023 13:50:54 +0200 Subject: [PATCH] Improve transformers versions handling (#2104) --- src/diffusers/pipelines/stable_diffusion/__init__.py | 2 +- src/diffusers/utils/import_utils.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/__init__.py b/src/diffusers/pipelines/stable_diffusion/__init__.py index 07e24d0cfd..71b87467d0 100644 --- a/src/diffusers/pipelines/stable_diffusion/__init__.py +++ b/src/diffusers/pipelines/stable_diffusion/__init__.py @@ -57,7 +57,7 @@ else: try: - if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.26.0.dev0")): + 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 diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index c87658763c..d87255e079 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -392,10 +392,10 @@ def requires_backends(obj, backends): if name in [ "StableDiffusionDepth2ImgPipeline", - ] and is_transformers_version("<", "4.26.0.dev0"): + ] and is_transformers_version("<", "4.26.0"): 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```" + f"You need to install `transformers>=4.26` in order to use {name}: \n```\n pip install" + " --upgrade transformers \n```" )