






'''simple docstring'''







import logging

from transformers.configuration_utils import PretrainedConfig


_A			:    Dict					  =logging.getLogger(__name__)

class _lowercase			(       _lowercase	):

							'''simple docstring'''





							a          = """masked_bert"""




							def __init__(							self:       Dict    ,							UpperCamelCase__:       int=30_522    ,							UpperCamelCase__:       Union[str, Any]=768    ,							UpperCamelCase__:       Tuple=12    ,							UpperCamelCase__:       Dict=12    ,							UpperCamelCase__:       List[Any]=3_072    ,							UpperCamelCase__:       Optional[Any]="gelu"    ,							UpperCamelCase__:       List[str]=0.1    ,							UpperCamelCase__:       int=0.1    ,							UpperCamelCase__:       Dict=512    ,							UpperCamelCase__:       List[Any]=2    ,							UpperCamelCase__:       Union[str, Any]=0.02    ,							UpperCamelCase__:       Dict=1e-12    ,							UpperCamelCase__:       Dict=0    ,							UpperCamelCase__:       Tuple="topK"    ,							UpperCamelCase__:       int="constant"    ,							UpperCamelCase__:       Dict=0.0    ,							**UpperCamelCase__:       List[str]    ,							):
														super().__init__(pad_token_id=UpperCamelCase__    ,							**UpperCamelCase__  )

														lowerCamelCase__      :			List[str] 						= vocab_size
														lowerCamelCase__      :			int 						= hidden_size
														lowerCamelCase__      :			Union[str, Any] 						= num_hidden_layers
														lowerCamelCase__      :			List[Any] 						= num_attention_heads
														lowerCamelCase__      :			str 						= hidden_act
														lowerCamelCase__      :			Union[str, Any] 						= intermediate_size
														lowerCamelCase__      :			int 						= hidden_dropout_prob
														lowerCamelCase__      :			Optional[int] 						= attention_probs_dropout_prob
														lowerCamelCase__      :			int 						= max_position_embeddings
														lowerCamelCase__      :			Tuple 						= type_vocab_size
														lowerCamelCase__      :			List[str] 						= initializer_range
														lowerCamelCase__      :			Any 						= layer_norm_eps
														lowerCamelCase__      :			Optional[int] 						= pruning_method
														lowerCamelCase__      :			Optional[int] 						= mask_init
														lowerCamelCase__      :			Tuple 						= mask_scale









'''simple docstring'''







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


_A			:    Optional[Any]					  =logging.get_logger(__name__)

_A			:    Optional[int]					  ={
    '''RWKV/rwkv-4-169m-pile''': '''https://huggingface.co/RWKV/rwkv-4-169m-pile/resolve/main/config.json''',
    '''RWKV/rwkv-4-430m-pile''': '''https://huggingface.co/RWKV/rwkv-4-430m-pile/resolve/main/config.json''',
    '''RWKV/rwkv-4-1b5-pile''': '''https://huggingface.co/RWKV/rwkv-4-1b5-pile/resolve/main/config.json''',
    '''RWKV/rwkv-4-3b-pile''': '''https://huggingface.co/RWKV/rwkv-4-3b-pile/resolve/main/config.json''',
    '''RWKV/rwkv-4-7b-pile''': '''https://huggingface.co/RWKV/rwkv-4-7b-pile/resolve/main/config.json''',
    '''RWKV/rwkv-4-14b-pile''': '''https://huggingface.co/RWKV/rwkv-4-14b-pile/resolve/main/config.json''',
    '''RWKV/rwkv-raven-1b5''': '''https://huggingface.co/RWKV/rwkv-raven-1b5/resolve/main/config.json''',
    '''RWKV/rwkv-raven-3b''': '''https://huggingface.co/RWKV/rwkv-raven-3b/resolve/main/config.json''',
    '''RWKV/rwkv-raven-7b''': '''https://huggingface.co/RWKV/rwkv-raven-7b/resolve/main/config.json''',
    '''RWKV/rwkv-raven-14b''': '''https://huggingface.co/RWKV/rwkv-raven-14b/resolve/main/config.json''',
}

class _lowercase			(       _lowercase	):
							a          = """rwkv"""
							a          = {"""max_position_embeddings""": """context_length"""}




							def __init__(							self:       Tuple    ,							UpperCamelCase__:       Optional[Any]=50_277    ,							UpperCamelCase__:       Union[str, Any]=1_024    ,							UpperCamelCase__:       Tuple=4_096    ,							UpperCamelCase__:       List[Any]=32    ,							UpperCamelCase__:       Dict=None    ,							UpperCamelCase__:       Dict=None    ,							UpperCamelCase__:       int=1e-5    ,							UpperCamelCase__:       Any=0    ,							UpperCamelCase__:       str=0    ,							UpperCamelCase__:       Union[str, Any]=6    ,							UpperCamelCase__:       Optional[int]=False    ,							UpperCamelCase__:       Dict=True    ,							**UpperCamelCase__:       Dict    ,							):
														lowerCamelCase__      :			Dict 						= vocab_size
														lowerCamelCase__      :			Optional[Any] 						= context_length
														lowerCamelCase__      :			Optional[Any] 						= hidden_size
														lowerCamelCase__      :			Any 						= num_hidden_layers
														lowerCamelCase__      :			int 						= attention_hidden_size if attention_hidden_size is not None else hidden_size
														lowerCamelCase__      :			Union[str, Any] 						= intermediate_size if intermediate_size is not None else 4 * hidden_size
														lowerCamelCase__      :			List[str] 						= layer_norm_epsilon
														lowerCamelCase__      :			int 						= rescale_every
														lowerCamelCase__      :			Optional[int] 						= use_cache

														lowerCamelCase__      :			Dict 						= bos_token_id
														lowerCamelCase__      :			Any 						= eos_token_id

														super().__init__(
														    tie_word_embeddings=UpperCamelCase__    ,							bos_token_id=UpperCamelCase__    ,							eos_token_id=UpperCamelCase__    ,							**UpperCamelCase__  )

