__init__.py 635 Bytes
Newer Older
1
2
3
4
5
6
from ...utils import (
    OptionalDependencyNotAvailable,
    is_torch_available,
    is_transformers_available,
    is_transformers_version,
)
Will Berman's avatar
Will Berman committed
7
8


9
10
11
12
try:
    if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
13
    from ...utils.dummy_torch_and_transformers_objects import UnCLIPImageVariationPipeline, UnCLIPPipeline
14
else:
Will Berman's avatar
Will Berman committed
15
    from .pipeline_unclip import UnCLIPPipeline
Will Berman's avatar
Will Berman committed
16
    from .pipeline_unclip_image_variation import UnCLIPImageVariationPipeline
17
    from .text_proj import UnCLIPTextProjModel