


from typing import TYPE_CHECKING

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


UpperCAmelCase_      : Tuple  					=       {
    'configuration_mega': ['MEGA_PRETRAINED_CONFIG_ARCHIVE_MAP', 'MegaConfig', 'MegaOnnxConfig'],
}

try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   UpperCAmelCase_      : List[Any]  					=       [
       'MEGA_PRETRAINED_MODEL_ARCHIVE_LIST',
       'MegaForCausalLM',
       'MegaForMaskedLM',
       'MegaForMultipleChoice',
       'MegaForQuestionAnswering',
       'MegaForSequenceClassification',
       'MegaForTokenClassification',
       'MegaModel',
       'MegaPreTrainedModel',
   ]

if TYPE_CHECKING:
   from .configuration_mega import MEGA_PRETRAINED_CONFIG_ARCHIVE_MAP, MegaConfig, MegaOnnxConfig

   try:
      if not is_torch_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_mega import (
          MEGA_PRETRAINED_MODEL_ARCHIVE_LIST,
          MegaForCausalLM,
          MegaForMaskedLM,
          MegaForMultipleChoice,
          MegaForQuestionAnswering,
          MegaForSequenceClassification,
          MegaForTokenClassification,
          MegaModel,
          MegaPreTrainedModel,
      )

else:
   import sys

   UpperCAmelCase_      : Optional[int]  					=       _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)






"""simple docstring"""





import numpy as np


def    _snake_case  (  _snake_case :	np.array	):
      return 1 / (1 + np.exp(-vector	))


if __name__ == "__main__":
 import doctest

 doctest.testmod()


