Skip to content

Commit 4cb887e

Browse files
authored
Transformers version req for UnCLIP (huggingface#1766)
* Transformers version req for UnCLIP * add to the list
1 parent 9f657f1 commit 4cb887e

File tree

2 files changed

+13
-3
lines changed

2 files changed

+13
-3
lines changed

src/diffusers/pipelines/unclip/__init__.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,15 @@
1-
from ...utils import is_torch_available, is_transformers_available
1+
from ...utils import (
2+
OptionalDependencyNotAvailable,
3+
is_torch_available,
4+
is_transformers_available,
5+
is_transformers_version,
6+
)
27

38

4-
if is_transformers_available() and is_torch_available():
9+
try:
10+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
11+
raise OptionalDependencyNotAvailable()
12+
except OptionalDependencyNotAvailable:
13+
from ...utils.dummy_torch_and_transformers_objects import UnCLIPPipeline
14+
else:
515
from .pipeline_unclip import UnCLIPPipeline
6-
from .text_proj import UnCLIPTextProjModel

src/diffusers/utils/import_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -359,6 +359,7 @@ def requires_backends(obj, backends):
359359
"VersatileDiffusionPipeline",
360360
"VersatileDiffusionDualGuidedPipeline",
361361
"StableDiffusionImageVariationPipeline",
362+
"UnCLIPPipeline",
362363
] and is_transformers_version("<", "4.25.0"):
363364
raise ImportError(
364365
f"You need to install `transformers>=4.25` in order to use {name}: \n```\n pip install"

0 commit comments

Comments
 (0)