
from ...utils import (
    OptionalDependencyNotAvailable,
    is_torch_available,
    is_transformers_available,
    is_transformers_version,
)


try:
	if not (is_transformers_available() and is_torch_available()):
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	from ...utils.dummy_torch_and_transformers_objects import (
	    ImageTextPipelineOutput,
	    UniDiffuserPipeline,
	)
else:
	from .modeling_text_decoder import UniDiffuserTextDecoder
	from .modeling_uvit import UniDiffuserModel, UTransformeraDModel
	from .pipeline_unidiffuser import ImageTextPipelineOutput, UniDiffuserPipeline




from ..utils import is_flax_available, is_torch_available


if is_torch_available():
	from .autoencoder_kl import AutoencoderKL
	from .controlnet import ControlNetModel
	from .dual_transformer_ad import DualTransformeraDModel
	from .modeling_utils import ModelMixin
	from .prior_transformer import PriorTransformer
	from .ta_film_transformer import TaFilmDecoder
	from .transformer_ad import TransformeraDModel
	from .unet_ad import UNetaDModel
	from .unet_ad import UNetaDModel
	from .unet_ad_condition import UNetaDConditionModel
	from .unet_ad_condition import UNetaDConditionModel
	from .vq_model import VQModel

if is_flax_available():
	from .controlnet_flax import FlaxControlNetModel
	from .unet_ad_condition_flax import FlaxUNetaDConditionModel
	from .vae_flax import FlaxAutoencoderKL

