






from typing import TYPE_CHECKING

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


A_		:		Dict		     =			{
    'configuration_gpt_neo': ['GPT_NEO_PRETRAINED_CONFIG_ARCHIVE_MAP', 'GPTNeoConfig', 'GPTNeoOnnxConfig'],
}

try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   A_		:		Tuple		     =			[
       '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:
   A_		:		Dict		     =			[
       '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

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







from abc import ABC, abstractmethod
from argparse import ArgumentParser




class     _lowerCAmelCase(	UpperCAmelCase_ ):





       """simple docstring"""







       @staticmethod
       @abstractmethod
       def       _a   ( _lowerCamelCase							):
         raise NotImplementedError()





       @abstractmethod
       def       _a   ( self							):
         raise NotImplementedError()