

'''simple docstring'''




from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_a				: List[str] 							=						{'configuration_umt5': ['UMT5Config', 'UMT5OnnxConfig']}


try:
  if not is_torch_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  _a				: Dict 							=						[
      'UMT5EncoderModel',
      'UMT5ForConditionalGeneration',
      'UMT5ForQuestionAnswering',
      'UMT5Model',
      'UMT5PreTrainedModel',
  ]

if TYPE_CHECKING:
  from .configuration_umta import UMTaConfig, UMTaOnnxConfig

  try:
    if not is_torch_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_umta import (
        UMTaEncoderModel,
        UMTaForConditionalGeneration,
        UMTaForQuestionAnswering,
        UMTaModel,
        UMTaPreTrainedModel,
    )
else:
  import sys

  _a				: List[Any] 							=						_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)





'''simple docstring'''




_a				: Tuple 							=						'\n# Diffusers installation\n! pip install diffusers transformers datasets accelerate\n# To install from source instead of the last release, comment the command above and uncomment the following one.\n# ! pip install git+https://github.com/huggingface/diffusers.git\n'

_a				: Tuple 							=						[{'type': 'code', 'content': INSTALL_CONTENT}]


