

'''simple docstring'''






from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


lowerCAmelCase__      		=			{
    'configuration_megatron_bert': ['MEGATRON_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP', 'MegatronBertConfig'],
}

try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   lowerCAmelCase__      		=			[
       'MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST',
       'MegatronBertForCausalLM',
       'MegatronBertForMaskedLM',
       'MegatronBertForMultipleChoice',
       'MegatronBertForNextSentencePrediction',
       'MegatronBertForPreTraining',
       'MegatronBertForQuestionAnswering',
       'MegatronBertForSequenceClassification',
       'MegatronBertForTokenClassification',
       'MegatronBertModel',
       'MegatronBertPreTrainedModel',
   ]

if TYPE_CHECKING:
   from .configuration_megatron_bert import MEGATRON_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP, MegatronBertConfig

   try:
      if not is_torch_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_megatron_bert import (
          MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
          MegatronBertForCausalLM,
          MegatronBertForMaskedLM,
          MegatronBertForMultipleChoice,
          MegatronBertForNextSentencePrediction,
          MegatronBertForPreTraining,
          MegatronBertForQuestionAnswering,
          MegatronBertForSequenceClassification,
          MegatronBertForTokenClassification,
          MegatronBertModel,
          MegatronBertPreTrainedModel,
      )

else:
   import sys

   lowerCAmelCase__      		=			_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)




'''simple docstring'''






from typing import TYPE_CHECKING

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


lowerCAmelCase__      		=			{'''configuration_vit_mae''': ['''VIT_MAE_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''ViTMAEConfig''']}

try:
 if not is_torch_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 lowerCAmelCase__      		=			[
     '''VIT_MAE_PRETRAINED_MODEL_ARCHIVE_LIST''',
     '''ViTMAEForPreTraining''',
     '''ViTMAELayer''',
     '''ViTMAEModel''',
     '''ViTMAEPreTrainedModel''',
 ]

try:
 if not is_tf_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 lowerCAmelCase__      		=			[
     '''TFViTMAEForPreTraining''',
     '''TFViTMAEModel''',
     '''TFViTMAEPreTrainedModel''',
 ]

if TYPE_CHECKING:
 from .configuration_vit_mae import VIT_MAE_PRETRAINED_CONFIG_ARCHIVE_MAP, ViTMAEConfig

 try:
  if not is_torch_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .modeling_vit_mae import (
      VIT_MAE_PRETRAINED_MODEL_ARCHIVE_LIST,
      ViTMAEForPreTraining,
      ViTMAELayer,
      ViTMAEModel,
      ViTMAEPreTrainedModel,
  )

 try:
  if not is_tf_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .modeling_tf_vit_mae import TFViTMAEForPreTraining, TFViTMAEModel, TFViTMAEPreTrainedModel


else:
 import sys

 lowerCAmelCase__      		=			_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)

