Skip to content

Commit 3ebe40f

Browse files
authored
Adapt to forced transformers version in some dependent libraries (huggingface#1638)
* Adapt to forced transformers version in some dependent libraries * style * Update __init__.py * update requires_backends
1 parent 0892525 commit 3ebe40f

File tree

4 files changed

+16
-15
lines changed

4 files changed

+16
-15
lines changed

src/diffusers/__init__.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -18,18 +18,6 @@
1818
)
1919

2020

21-
# Make sure `transformers` is up to date
22-
if is_transformers_available():
23-
import transformers
24-
25-
if is_transformers_version("<", "4.25.1"):
26-
raise ImportError(
27-
f"`diffusers` requires transformers >= 4.25.1 to function correctly, but {transformers.__version__} was"
28-
" found in your environment. You can upgrade it with pip: `pip install transformers --upgrade`"
29-
)
30-
else:
31-
pass
32-
3321
try:
3422
if not is_torch_available():
3523
raise OptionalDependencyNotAvailable()

src/diffusers/pipelines/stable_diffusion/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ class StableDiffusionPipelineOutput(BaseOutput):
4646
from .safety_checker import StableDiffusionSafetyChecker
4747

4848
try:
49-
if not (is_transformers_available() and is_torch_available()):
49+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
5050
raise OptionalDependencyNotAvailable()
5151
except OptionalDependencyNotAvailable:
5252
from ...utils.dummy_torch_and_transformers_objects import StableDiffusionImageVariationPipeline

src/diffusers/pipelines/versatile_diffusion/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88

99
try:
10-
if not (is_transformers_available() and is_torch_available()):
10+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
1111
raise OptionalDependencyNotAvailable()
1212
except OptionalDependencyNotAvailable:
1313
from ...utils.dummy_torch_and_transformers_objects import (

src/diffusers/utils/import_utils.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,20 @@ def requires_backends(obj, backends):
354354
if failed:
355355
raise ImportError("".join(failed))
356356

357-
if name in ["StableDiffusionDepth2ImgPipeline"] and is_transformers_version("<", "4.26.0.dev0"):
357+
if name in [
358+
"VersatileDiffusionTextToImagePipeline",
359+
"VersatileDiffusionPipeline",
360+
"VersatileDiffusionDualGuidedPipeline",
361+
"StableDiffusionImageVariationPipeline",
362+
] and is_transformers_version("<", "4.25.0"):
363+
raise ImportError(
364+
f"You need to install `transformers>=4.25` in order to use {name}: \n```\n pip install"
365+
" --upgrade transformers \n```"
366+
)
367+
368+
if name in [
369+
"StableDiffusionDepth2ImgPipeline",
370+
] and is_transformers_version("<", "4.26.0.dev0"):
358371
raise ImportError(
359372
f"You need to install `transformers` from 'main' in order to use {name}: \n```\n pip install"
360373
" git+https://github.com/huggingface/transformers \n```"

0 commit comments

Comments
 (0)