





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


UpperCAmelCase_					:   List[str] 							=						logging.get_logger(__name__)

# TODO: upload to AWS
UpperCAmelCase_					:   Any 							=						{
    """yjernite/retribert-base-uncased""": (
        """https://huggingface.co/yjernite/retribert-base-uncased/resolve/main/config.json"""
    ),
}





class      lowercase__    (							__A  ):
  __UpperCamelCase	     =  """retribert"""







  def __init__(	self    ,						_lowercase=30_522    ,						_lowercase=768    ,						_lowercase=8    ,						_lowercase=12    ,						_lowercase=3_072    ,						_lowercase="gelu"    ,						_lowercase=0.1    ,						_lowercase=0.1    ,						_lowercase=512    ,						_lowercase=2    ,						_lowercase=0.02    ,						_lowercase=1e-12    ,						_lowercase=True    ,						_lowercase=128    ,						_lowercase=0    ,						**_lowercase    ,						):
       super().__init__(pad_token_id=_lowercase    ,						**_lowercase						)

       lowerCAmelCase_					:       List[str]    =		vocab_size
       lowerCAmelCase_					:       Union[str, Any]    =		hidden_size
       lowerCAmelCase_					:       int    =		num_hidden_layers
       lowerCAmelCase_					:       Any    =		num_attention_heads
       lowerCAmelCase_					:       List[str]    =		hidden_act
       lowerCAmelCase_					:       str    =		intermediate_size
       lowerCAmelCase_					:       Optional[Any]    =		hidden_dropout_prob
       lowerCAmelCase_					:       int    =		attention_probs_dropout_prob
       lowerCAmelCase_					:       Dict    =		max_position_embeddings
       lowerCAmelCase_					:       List[Any]    =		type_vocab_size
       lowerCAmelCase_					:       str    =		initializer_range
       lowerCAmelCase_					:       Union[str, Any]    =		layer_norm_eps
       lowerCAmelCase_					:       List[str]    =		share_encoders
       lowerCAmelCase_					:       List[str]    =		projection_dim







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


UpperCAmelCase_					:   Tuple 							=						logging.get_logger(__name__)

UpperCAmelCase_					:   List[str] 							=						{
    """google/fnet-base""": """https://huggingface.co/google/fnet-base/resolve/main/config.json""",
    """google/fnet-large""": """https://huggingface.co/google/fnet-large/resolve/main/config.json"""
    # See all FNet models at https://huggingface.co/models?filter=fnet
}





class      lowercase__    (							__A  ):
  __UpperCamelCase	     =  """fnet"""







  def __init__(	self    ,						_lowercase=32_000    ,						_lowercase=768    ,						_lowercase=12    ,						_lowercase=3_072    ,						_lowercase="gelu_new"    ,						_lowercase=0.1    ,						_lowercase=512    ,						_lowercase=4    ,						_lowercase=0.02    ,						_lowercase=1e-12    ,						_lowercase=False    ,						_lowercase=512    ,						_lowercase=3    ,						_lowercase=1    ,						_lowercase=2    ,						**_lowercase    ,						):
       super().__init__(pad_token_id=_lowercase    ,						bos_token_id=_lowercase    ,						eos_token_id=_lowercase    ,						**_lowercase						)

       lowerCAmelCase_					:       Optional[Any]    =		vocab_size
       lowerCAmelCase_					:       List[str]    =		max_position_embeddings
       lowerCAmelCase_					:       str    =		hidden_size
       lowerCAmelCase_					:       Any    =		num_hidden_layers
       lowerCAmelCase_					:       Optional[int]    =		intermediate_size
       lowerCAmelCase_					:       int    =		hidden_act
       lowerCAmelCase_					:       List[str]    =		hidden_dropout_prob
       lowerCAmelCase_					:       Tuple    =		initializer_range
       lowerCAmelCase_					:       int    =		type_vocab_size
       lowerCAmelCase_					:       str    =		layer_norm_eps
       lowerCAmelCase_					:       Optional[Any]    =		use_tpu_fourier_optimizations
       lowerCAmelCase_					:       Union[str, Any]    =		tpu_short_seq_length
