

'''simple docstring'''

from ...configuration_utils import PretrainedConfig
from ...utils import logging


lowercase__            =     logging.get_logger(__name__)

lowercase__            =     {
    """facebook/regnet-y-040""": """https://huggingface.co/facebook/regnet-y-040/blob/main/config.json""",
}




class 			A_					(							_lowerCamelCase				):






     '''simple docstring'''



     UpperCAmelCase_		:      Union[str, Any]    						=    """regnet"""
     UpperCAmelCase_		:      Dict    						=    ["""x""", """y"""]


     def __init__(						self					:	List[Any]	,     lowercase_					:	List[Any]=3	,     lowercase_					:	Union[str, Any]=32	,     lowercase_					:	Optional[Any]=[128, 192, 512, 1_088]	,     lowercase_					:	int=[2, 6, 12, 2]	,     lowercase_					:	int=64	,     lowercase_					:	Optional[Any]="y"	,     lowercase_					:	List[str]="relu"	,     **lowercase_					:	Any	,     )		->      List[Any]:
      super().__init__(**__lowerCamelCase			)
      if layer_type not in self.layer_types:
       raise ValueError(f"""layer_type={layer_type} is not one of {",".join(self.layer_types			)}"""			)
      UpperCAmelCase			:      Dict		   =					num_channels
      UpperCAmelCase			:      List[str]		   =					embedding_size
      UpperCAmelCase			:      Optional[Any]		   =					hidden_sizes
      UpperCAmelCase			:      List[str]		   =					depths
      UpperCAmelCase			:      List[str]		   =					groups_width
      UpperCAmelCase			:      Any		   =					layer_type
      UpperCAmelCase			:      Optional[Any]		   =					hidden_act
      # always downsample in the first stage
      UpperCAmelCase			:      List[Any]		   =					True












'''simple docstring'''





from typing import TYPE_CHECKING

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


__A				:				Union[str, Any]					=		{
    """configuration_gpt2""": ["""GPT2_PRETRAINED_CONFIG_ARCHIVE_MAP""", """GPT2Config""", """GPT2OnnxConfig"""],
    """tokenization_gpt2""": ["""GPT2Tokenizer"""],
}

try:
 if not is_tokenizers_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 __A				:				Tuple					=		["""GPT2TokenizerFast"""]

try:
 if not is_torch_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 __A				:				Optional[int]					=		[
     """GPT2_PRETRAINED_MODEL_ARCHIVE_LIST""",
     """GPT2DoubleHeadsModel""",
     """GPT2ForQuestionAnswering""",
     """GPT2ForSequenceClassification""",
     """GPT2ForTokenClassification""",
     """GPT2LMHeadModel""",
     """GPT2Model""",
     """GPT2PreTrainedModel""",
     """load_tf_weights_in_gpt2""",
 ]

try:
 if not is_tf_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 __A				:				Optional[Any]					=		[
     """TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST""",
     """TFGPT2DoubleHeadsModel""",
     """TFGPT2ForSequenceClassification""",
     """TFGPT2LMHeadModel""",
     """TFGPT2MainLayer""",
     """TFGPT2Model""",
     """TFGPT2PreTrainedModel""",
 ]

try:
 if not is_keras_nlp_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 __A				:				Any					=		["""TFGPT2Tokenizer"""]

try:
 if not is_flax_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 __A				:				Tuple					=		["""FlaxGPT2LMHeadModel""", """FlaxGPT2Model""", """FlaxGPT2PreTrainedModel"""]

if TYPE_CHECKING:
 from .configuration_gpta import GPT2_PRETRAINED_CONFIG_ARCHIVE_MAP, GPTaConfig, GPTaOnnxConfig
 from .tokenization_gpta import GPTaTokenizer

 try:
  if not is_tokenizers_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .tokenization_gpta_fast import GPTaTokenizerFast

 try:
  if not is_torch_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .modeling_gpta import (
      GPT2_PRETRAINED_MODEL_ARCHIVE_LIST,
      GPTaDoubleHeadsModel,
      GPTaForQuestionAnswering,
      GPTaForSequenceClassification,
      GPTaForTokenClassification,
      GPTaLMHeadModel,
      GPTaModel,
      GPTaPreTrainedModel,
      load_tf_weights_in_gpta,
  )

 try:
  if not is_tf_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .modeling_tf_gpta import (
      TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST,
      TFGPTaDoubleHeadsModel,
      TFGPTaForSequenceClassification,
      TFGPTaLMHeadModel,
      TFGPTaMainLayer,
      TFGPTaModel,
      TFGPTaPreTrainedModel,
  )

 try:
  if not is_keras_nlp_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .tokenization_gpta_tf import TFGPTaTokenizer

 try:
  if not is_flax_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .modeling_flax_gpta import FlaxGPTaLMHeadModel, FlaxGPTaModel, FlaxGPTaPreTrainedModel

else:
 import sys

 __A				:				Tuple					=		_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)

