


from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_sentencepiece_available,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
)


if is_sentencepiece_available():
  from ..ta.tokenization_ta import TaTokenizer
else:
  from ...utils.dummy_sentencepiece_objects import TaTokenizer

UpperCAmelCase__   	=							TaTokenizer

if is_tokenizers_available():
  from ..ta.tokenization_ta_fast import TaTokenizerFast
else:
  from ...utils.dummy_tokenizers_objects import TaTokenizerFast

UpperCAmelCase__   	=							TaTokenizerFast

UpperCAmelCase__   	=							{'''configuration_mt5''': ['''MT5Config''', '''MT5OnnxConfig''']}

try:
  if not is_torch_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  UpperCAmelCase__   	=							[
      '''MT5EncoderModel''',
      '''MT5ForConditionalGeneration''',
      '''MT5ForQuestionAnswering''',
      '''MT5Model''',
      '''MT5PreTrainedModel''',
      '''MT5Stack''',
  ]

try:
  if not is_tf_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  UpperCAmelCase__   	=							['''TFMT5EncoderModel''', '''TFMT5ForConditionalGeneration''', '''TFMT5Model''']

try:
  if not is_flax_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  UpperCAmelCase__   	=							['''FlaxMT5EncoderModel''', '''FlaxMT5ForConditionalGeneration''', '''FlaxMT5Model''']


if TYPE_CHECKING:
  from .configuration_mta import MTaConfig, MTaOnnxConfig

  try:
    if not is_torch_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_mta import (
        MTaEncoderModel,
        MTaForConditionalGeneration,
        MTaForQuestionAnswering,
        MTaModel,
        MTaPreTrainedModel,
        MTaStack,
    )

  try:
    if not is_tf_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_tf_mta import TFMTaEncoderModel, TFMTaForConditionalGeneration, TFMTaModel

  try:
    if not is_flax_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_flax_mta import FlaxMTaEncoderModel, FlaxMTaForConditionalGeneration, FlaxMTaModel

else:
  import sys

  UpperCAmelCase__   	=							_LazyModule(
      __name__,
      globals()['''__file__'''],
      _import_structure,
      extra_objects={'''MT5Tokenizer''': MTaTokenizer, '''MT5TokenizerFast''': MTaTokenizerFast},
      module_spec=__spec__,
  )





# flake8: noqa
# Lint as: python3
UpperCAmelCase__   	=							[
    '''VerificationMode''',
    '''Version''',
    '''disable_progress_bar''',
    '''enable_progress_bar''',
    '''is_progress_bar_enabled''',
    '''experimental''',
]

from .info_utils import VerificationMode
from .logging import disable_progress_bar, enable_progress_bar, is_progress_bar_enabled
from .version import Version
from .experimental import experimental

