




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


a_      						=   logging.get_logger(__name__)

a_      						=   {
    """openbmb/cpm-ant-10b""": """https://huggingface.co/openbmb/cpm-ant-10b/blob/main/config.json"""
    # See all CPMAnt models at https://huggingface.co/models?filter=cpmant
}

class        _lowerCamelCase  (	_SCREAMING_SNAKE_CASE      ):

      """simple docstring"""

      lowerCAmelCase__	:       List[Any]			=    "cpmant"







      def __init__(   self    :       Dict       ,				snake_case    :       str = 30720       ,				snake_case    :       int = 4096       ,				snake_case    :       int = 32       ,				snake_case    :       Dict = 128       ,				snake_case    :       List[Any] = 10240       ,				snake_case    :       List[str] = 48       ,				snake_case    :       Any = 0.0       ,				snake_case    :       Union[str, Any] = 512       ,				snake_case    :       Optional[Any] = 2048       ,				snake_case    :       str = 1E-6       ,				snake_case    :       Optional[Any] = 1.0       ,				snake_case    :       Dict = 32       ,				snake_case    :       int = 32       ,				snake_case    :       List[Any] = 32       ,				snake_case    :       Dict = True       ,				**snake_case    :       str       ,				):
          super().__init__(**UpperCAmelCase_ )
          __UpperCamelCase     						=  prompt_types
          __UpperCamelCase     						=  prompt_length
          __UpperCamelCase     						=  segment_types
          __UpperCamelCase     						=  hidden_size
          __UpperCamelCase     						=  num_attention_heads
          __UpperCamelCase     						=  dim_head
          __UpperCamelCase     						=  dim_ff
          __UpperCamelCase     						=  num_hidden_layers
          __UpperCamelCase     						=  position_bias_num_buckets
          __UpperCamelCase     						=  position_bias_max_distance
          __UpperCamelCase     						=  dropout_p
          __UpperCamelCase     						=  eps
          __UpperCamelCase     						=  use_cache
          __UpperCamelCase     						=  vocab_size
          __UpperCamelCase     						=  init_std











"""simple docstring"""







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


UpperCAmelCase:   Dict       			=							logging.get_logger(__name__)

UpperCAmelCase:   Optional[Any]       			=							{
    """facebook/opt-125m""": """https://huggingface.co/facebook/opt-125m/blob/main/config.json""",
    """facebook/opt-350m""": """https://huggingface.co/facebook/opt-350m/blob/main/config.json""",
    """facebook/opt-1.3b""": """https://huggingface.co/facebook/opt-1.3b/blob/main/config.json""",
    """facebook/opt-2.7b""": """https://huggingface.co/facebook/opt-2.7b/blob/main/config.json""",
    """facebook/opt-6.7b""": """https://huggingface.co/facebook/opt-6.7b/blob/main/config.json""",
    """facebook/opt-13b""": """https://huggingface.co/facebook/opt-13b/blob/main/config.json""",
}





class    UpperCamelCase     (			snake_case    ):




	"""simple docstring"""
	SCREAMING_SNAKE_CASE_				:    Dict		    =   "opt"
	SCREAMING_SNAKE_CASE_				:    List[str]		    =   ["past_key_values"]






	def __init__(			self							,UpperCAmelCase_=5_02_72							,UpperCAmelCase_=7_68							,UpperCAmelCase_=12							,UpperCAmelCase_=30_72							,UpperCAmelCase_=20_48							,UpperCAmelCase_=True							,UpperCAmelCase_=False							,UpperCAmelCase_=None							,UpperCAmelCase_=0.1							,UpperCAmelCase_=0.0							,UpperCAmelCase_=12							,UpperCAmelCase_="relu"							,UpperCAmelCase_=0.0							,UpperCAmelCase_=0.02							,UpperCAmelCase_=True							,UpperCAmelCase_=1							,UpperCAmelCase_=2							,UpperCAmelCase_=2							,UpperCAmelCase_=True							,UpperCAmelCase_=True							,**UpperCAmelCase_							,):
			super().__init__(
			    pad_token_id=UpperCAmelCase_							,bos_token_id=UpperCAmelCase_							,eos_token_id=UpperCAmelCase_							,**UpperCAmelCase_							,)
			_lowercase							:      Optional[Any]			=					vocab_size
			_lowercase							:      Tuple			=					max_position_embeddings
			_lowercase							:      Union[str, Any]			=					num_attention_heads
			_lowercase							:      List[str]			=					word_embed_proj_dim if word_embed_proj_dim is not None else hidden_size
			_lowercase							:      List[str]			=					ffn_dim
			_lowercase							:      List[str]			=					hidden_size
			_lowercase							:      List[Any]			=					num_hidden_layers
			_lowercase							:      List[Any]			=					dropout
			_lowercase							:      str			=					attention_dropout
			_lowercase							:      List[Any]			=					activation_function
			_lowercase							:      int			=					init_std
			_lowercase							:      List[Any]			=					layerdrop
			_lowercase							:      Tuple			=					use_cache
			_lowercase							:      Any			=					do_layer_norm_before
			# We keep these variables at `True` for backward compatibility.
			_lowercase							:      int			=					enable_bias
			_lowercase							:      str			=					layer_norm_elementwise_affine

			# Note that the only purpose of `_remove_final_layer_norm` is to keep backward compatibility
			# with checkpoints that have been fine-tuned before transformers v4.20.1
			# see https://github.com/facebookresearch/metaseq/pull/164
			_lowercase							:      Optional[Any]			=					_remove_final_layer_norm



