






from typing import TYPE_CHECKING

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


UpperCAmelCase__				   =		{
    "configuration_electra": ["ELECTRA_PRETRAINED_CONFIG_ARCHIVE_MAP", "ElectraConfig", "ElectraOnnxConfig"],
    "tokenization_electra": ["ElectraTokenizer"],
}

try:
     if not is_tokenizers_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     UpperCAmelCase__				   =		["ElectraTokenizerFast"]

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     UpperCAmelCase__				   =		[
         "ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST",
         "ElectraForCausalLM",
         "ElectraForMaskedLM",
         "ElectraForMultipleChoice",
         "ElectraForPreTraining",
         "ElectraForQuestionAnswering",
         "ElectraForSequenceClassification",
         "ElectraForTokenClassification",
         "ElectraModel",
         "ElectraPreTrainedModel",
         "load_tf_weights_in_electra",
     ]

try:
     if not is_tf_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     UpperCAmelCase__				   =		[
         "TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST",
         "TFElectraForMaskedLM",
         "TFElectraForMultipleChoice",
         "TFElectraForPreTraining",
         "TFElectraForQuestionAnswering",
         "TFElectraForSequenceClassification",
         "TFElectraForTokenClassification",
         "TFElectraModel",
         "TFElectraPreTrainedModel",
     ]

try:
     if not is_flax_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     UpperCAmelCase__				   =		[
         "FlaxElectraForCausalLM",
         "FlaxElectraForMaskedLM",
         "FlaxElectraForMultipleChoice",
         "FlaxElectraForPreTraining",
         "FlaxElectraForQuestionAnswering",
         "FlaxElectraForSequenceClassification",
         "FlaxElectraForTokenClassification",
         "FlaxElectraModel",
         "FlaxElectraPreTrainedModel",
     ]


if TYPE_CHECKING:
     from .configuration_electra import ELECTRA_PRETRAINED_CONFIG_ARCHIVE_MAP, ElectraConfig, ElectraOnnxConfig
     from .tokenization_electra import ElectraTokenizer

     try:
          if not is_tokenizers_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .tokenization_electra_fast import ElectraTokenizerFast

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_electra import (
              ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST,
              ElectraForCausalLM,
              ElectraForMaskedLM,
              ElectraForMultipleChoice,
              ElectraForPreTraining,
              ElectraForQuestionAnswering,
              ElectraForSequenceClassification,
              ElectraForTokenClassification,
              ElectraModel,
              ElectraPreTrainedModel,
              load_tf_weights_in_electra,
          )

     try:
          if not is_tf_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_tf_electra import (
              TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST,
              TFElectraForMaskedLM,
              TFElectraForMultipleChoice,
              TFElectraForPreTraining,
              TFElectraForQuestionAnswering,
              TFElectraForSequenceClassification,
              TFElectraForTokenClassification,
              TFElectraModel,
              TFElectraPreTrainedModel,
          )

     try:
          if not is_flax_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_flax_electra import (
              FlaxElectraForCausalLM,
              FlaxElectraForMaskedLM,
              FlaxElectraForMultipleChoice,
              FlaxElectraForPreTraining,
              FlaxElectraForQuestionAnswering,
              FlaxElectraForSequenceClassification,
              FlaxElectraForTokenClassification,
              FlaxElectraModel,
              FlaxElectraPreTrainedModel,
          )

else:
     import sys

     UpperCAmelCase__				   =		_LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)










"""simple docstring"""
from typing import Mapping

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxSeqaSeqConfigWithPast
from ...utils import logging


lowerCamelCase_					=      logging.get_logger(__name__)

lowerCamelCase_					=      {
    '''t5-small''': '''https://huggingface.co/t5-small/resolve/main/config.json''',
    '''t5-base''': '''https://huggingface.co/t5-base/resolve/main/config.json''',
    '''t5-large''': '''https://huggingface.co/t5-large/resolve/main/config.json''',
    '''t5-3b''': '''https://huggingface.co/t5-3b/resolve/main/config.json''',
    '''t5-11b''': '''https://huggingface.co/t5-11b/resolve/main/config.json''',
}







class 			UpperCamelCase_       (__A     ):
	__magic_name__	      =				'''t5'''
	__magic_name__	      =				['''past_key_values''']
	__magic_name__	      =				{'''hidden_size''': '''d_model''', '''num_attention_heads''': '''num_heads''', '''num_hidden_layers''': '''num_layers'''}





	def __init__(      self					: str	,	lowerCAmelCase_					: List[Any]=32_128	,	lowerCAmelCase_					: Tuple=512	,	lowerCAmelCase_					: Optional[int]=64	,	lowerCAmelCase_					: List[str]=2_048	,	lowerCAmelCase_					: Tuple=6	,	lowerCAmelCase_					: Dict=None	,	lowerCAmelCase_					: str=8	,	lowerCAmelCase_					: Optional[int]=32	,	lowerCAmelCase_					: Dict=128	,	lowerCAmelCase_					: Optional[Any]=0.1	,	lowerCAmelCase_					: str=1e-6	,	lowerCAmelCase_					: Dict=1.0	,	lowerCAmelCase_					: str="relu"	,	lowerCAmelCase_					: List[Any]=True	,	lowerCAmelCase_					: List[str]=True	,	lowerCAmelCase_					: Any=0	,	lowerCAmelCase_					: Tuple=1	,	**lowerCAmelCase_					: Optional[int]	,	)						->			int:
				UpperCAmelCase_     : int			     =					vocab_size
				UpperCAmelCase_     : Optional[Any]			     =					d_model
				UpperCAmelCase_     : str			     =					d_kv
				UpperCAmelCase_     : Any			     =					d_ff
				UpperCAmelCase_     : int			     =					num_layers
				UpperCAmelCase_     : Union[str, Any]			     =					(
				    num_decoder_layers if num_decoder_layers is not None else self.num_layers
				)  # default = symmetry
				UpperCAmelCase_     : Optional[Any]			     =					num_heads
				UpperCAmelCase_     : Any			     =					relative_attention_num_buckets
				UpperCAmelCase_     : Optional[Any]			     =					relative_attention_max_distance
				UpperCAmelCase_     : Optional[Any]			     =					dropout_rate
				UpperCAmelCase_     : Tuple			     =					layer_norm_epsilon
				UpperCAmelCase_     : int			     =					initializer_factor
				UpperCAmelCase_     : int			     =					feed_forward_proj
				UpperCAmelCase_     : str			     =					use_cache

				UpperCAmelCase_     : Tuple			     =					self.feed_forward_proj.split("-"						)
				UpperCAmelCase_     : List[Any]			     =					act_info[-1]
				UpperCAmelCase_     : Optional[int]			     =					act_info[0] == "gated"

				if len(lowerCAmelCase_						) > 1 and act_info[0] != "gated" or len(lowerCAmelCase_						) > 2:
							raise ValueError(
							    f"""`feed_forward_proj`: {feed_forward_proj} is not a valid activation function of the dense layer."""
							    "Please make sure `feed_forward_proj` is of the format `gated-{ACT_FN}` or `{ACT_FN}`, e.g. "
							    "'gated-gelu' or 'relu'"						)

				# for backwards compatibility
				if feed_forward_proj == "gated-gelu":
							UpperCAmelCase_     : int			     =					"gelu_new"

				super().__init__(
				    pad_token_id=lowerCAmelCase_	,	eos_token_id=lowerCAmelCase_	,	is_encoder_decoder=lowerCAmelCase_	,	**lowerCAmelCase_	,	)







class 			UpperCamelCase_       (__A     ):





	@property
	def      _SCREAMING_SNAKE_CASE      (      self					: Dict						)						->			Mapping[str, Mapping[int, str]]:
				UpperCAmelCase_     : Any			     =					{
				    "input_ids": {0: "batch", 1: "encoder_sequence"},
				    "attention_mask": {0: "batch", 1: "encoder_sequence"},
				}
				if self.use_past:
							UpperCAmelCase_     : List[Any]			     =					"past_encoder_sequence + sequence"
							UpperCAmelCase_     : Union[str, Any]			     =					{0: "batch"}
							UpperCAmelCase_     : Optional[Any]			     =					{0: "batch", 1: "past_decoder_sequence + sequence"}
				else:
							UpperCAmelCase_     : List[Any]			     =					{0: "batch", 1: "decoder_sequence"}
							UpperCAmelCase_     : Tuple			     =					{0: "batch", 1: "decoder_sequence"}

				if self.use_past:
							self.fill_with_past_key_values_(lowerCAmelCase_	,	direction="inputs"						)

				return common_inputs





	@property
	def      _SCREAMING_SNAKE_CASE      (      self					: Any						)						->			int:
				return 13



