Unverified Commit 4cb887e0 authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Transformers version req for UnCLIP (#1766)

* Transformers version req for UnCLIP

* add to the list
parent 9f657f10
from ...utils import is_torch_available, is_transformers_available
from ...utils import (
OptionalDependencyNotAvailable,
is_torch_available,
is_transformers_available,
is_transformers_version,
)
if is_transformers_available() and is_torch_available():
try:
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
from ...utils.dummy_torch_and_transformers_objects import UnCLIPPipeline
else:
from .pipeline_unclip import UnCLIPPipeline
from .text_proj import UnCLIPTextProjModel
......@@ -359,6 +359,7 @@ def requires_backends(obj, backends):
"VersatileDiffusionPipeline",
"VersatileDiffusionDualGuidedPipeline",
"StableDiffusionImageVariationPipeline",
"UnCLIPPipeline",
] and is_transformers_version("<", "4.25.0"):
raise ImportError(
f"You need to install `transformers>=4.25` in order to use {name}: \n```\n pip install"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment