





'''simple docstring'''

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_flax_available, is_torch_available


UpperCAmelCase					:		List[str]		=							{
    'configuration_gpt_neo': ['GPT_NEO_PRETRAINED_CONFIG_ARCHIVE_MAP', 'GPTNeoConfig', 'GPTNeoOnnxConfig'],
}

try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   UpperCAmelCase					:		int		=							[
       'GPT_NEO_PRETRAINED_MODEL_ARCHIVE_LIST',
       'GPTNeoForCausalLM',
       'GPTNeoForQuestionAnswering',
       'GPTNeoForSequenceClassification',
       'GPTNeoForTokenClassification',
       'GPTNeoModel',
       'GPTNeoPreTrainedModel',
       'load_tf_weights_in_gpt_neo',
   ]

try:
   if not is_flax_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   UpperCAmelCase					:		List[Any]		=							[
       'FlaxGPTNeoForCausalLM',
       'FlaxGPTNeoModel',
       'FlaxGPTNeoPreTrainedModel',
   ]


if TYPE_CHECKING:
   from .configuration_gpt_neo import GPT_NEO_PRETRAINED_CONFIG_ARCHIVE_MAP, GPTNeoConfig, GPTNeoOnnxConfig

   try:
      if not is_torch_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_gpt_neo import (
          GPT_NEO_PRETRAINED_MODEL_ARCHIVE_LIST,
          GPTNeoForCausalLM,
          GPTNeoForQuestionAnswering,
          GPTNeoForSequenceClassification,
          GPTNeoForTokenClassification,
          GPTNeoModel,
          GPTNeoPreTrainedModel,
          load_tf_weights_in_gpt_neo,
      )

   try:
      if not is_flax_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_flax_gpt_neo import FlaxGPTNeoForCausalLM, FlaxGPTNeoModel, FlaxGPTNeoPreTrainedModel


else:
   import sys

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









'''simple docstring'''

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


UpperCAmelCase					:		int		=							{
    'configuration_pegasus_x': ['PEGASUS_X_PRETRAINED_CONFIG_ARCHIVE_MAP', 'PegasusXConfig'],
}

try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   UpperCAmelCase					:		Union[str, Any]		=							[
       'PEGASUS_X_PRETRAINED_MODEL_ARCHIVE_LIST',
       'PegasusXForConditionalGeneration',
       'PegasusXModel',
       'PegasusXPreTrainedModel',
   ]


if TYPE_CHECKING:
   from .configuration_pegasus_x import PEGASUS_X_PRETRAINED_CONFIG_ARCHIVE_MAP, PegasusXConfig

   try:
      if not is_torch_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_pegasus_x import (
          PEGASUS_X_PRETRAINED_MODEL_ARCHIVE_LIST,
          PegasusXForConditionalGeneration,
          PegasusXModel,
          PegasusXPreTrainedModel,
      )


else:
   import sys

   UpperCAmelCase					:		Dict		=							_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)


