|
from ...utils import ( |
|
OptionalDependencyNotAvailable, |
|
is_torch_available, |
|
is_transformers_available, |
|
is_transformers_version, |
|
) |
|
|
|
|
|
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 ( |
|
VersatileDiffusionDualGuidedPipeline, |
|
VersatileDiffusionImageVariationPipeline, |
|
VersatileDiffusionPipeline, |
|
VersatileDiffusionTextToImagePipeline, |
|
) |
|
else: |
|
from .modeling_text_unet import UNetFlatConditionModel |
|
from .pipeline_versatile_diffusion import VersatileDiffusionPipeline |
|
from .pipeline_versatile_diffusion_dual_guided import VersatileDiffusionDualGuidedPipeline |
|
from .pipeline_versatile_diffusion_image_variation import VersatileDiffusionImageVariationPipeline |
|
from .pipeline_versatile_diffusion_text_to_image import VersatileDiffusionTextToImagePipeline |
|
|