


'''simple docstring'''





from collections import OrderedDict
from typing import Mapping

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


UpperCAmelCase__  			=				logging.get_logger(__name__)

UpperCAmelCase__  			=				{
    '''distilbert-base-uncased''': '''https://huggingface.co/distilbert-base-uncased/resolve/main/config.json''',
    '''distilbert-base-uncased-distilled-squad''': (
        '''https://huggingface.co/distilbert-base-uncased-distilled-squad/resolve/main/config.json'''
    ),
    '''distilbert-base-cased''': '''https://huggingface.co/distilbert-base-cased/resolve/main/config.json''',
    '''distilbert-base-cased-distilled-squad''': (
        '''https://huggingface.co/distilbert-base-cased-distilled-squad/resolve/main/config.json'''
    ),
    '''distilbert-base-german-cased''': '''https://huggingface.co/distilbert-base-german-cased/resolve/main/config.json''',
    '''distilbert-base-multilingual-cased''': (
        '''https://huggingface.co/distilbert-base-multilingual-cased/resolve/main/config.json'''
    ),
    '''distilbert-base-uncased-finetuned-sst-2-english''': (
        '''https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english/resolve/main/config.json'''
    ),
}
class 						a__					(					a_   ):





			'''simple docstring'''

			A						:				List[Any]					      =			'''distilbert'''
			A						:				int					      =			{
			    '''hidden_size''': '''dim''',
			    '''num_attention_heads''': '''n_heads''',
			    '''num_hidden_layers''': '''n_layers''',
			}
			def __init__(  self							:			Tuple			,    lowerCAmelCase_							:			Tuple=30_522			,    lowerCAmelCase_							:			Any=512			,    lowerCAmelCase_							:			Optional[Any]=False			,    lowerCAmelCase_							:			Dict=6			,    lowerCAmelCase_							:			str=12			,    lowerCAmelCase_							:			List[str]=768			,    lowerCAmelCase_							:			List[str]=4 * 768			,    lowerCAmelCase_							:			List[str]=0.1			,    lowerCAmelCase_							:			Dict=0.1			,    lowerCAmelCase_							:			int="gelu"			,    lowerCAmelCase_							:			List[Any]=0.02			,    lowerCAmelCase_							:			str=0.1			,    lowerCAmelCase_							:			Optional[Any]=0.2			,    lowerCAmelCase_							:			Optional[int]=0			,    **lowerCAmelCase_							:			Dict			,    )      ->		Dict:
						__A=	vocab_size
						__A=	max_position_embeddings
						__A=	sinusoidal_pos_embds
						__A=	n_layers
						__A=	n_heads
						__A=	dim
						__A=	hidden_dim
						__A=	dropout
						__A=	attention_dropout
						__A=	activation
						__A=	initializer_range
						__A=	qa_dropout
						__A=	seq_classif_dropout
						super().__init__(**lowerCAmelCase_			,    pad_token_id=lowerCAmelCase_	)


class 						a__					(					a_   ):





			'''simple docstring'''

			@property
			def 					lowerCAmelCase     (  self							:			Optional[Any]	)      ->		Mapping[str, Mapping[int, str]]:
						if self.task == "multiple-choice":
									__A=	{0: 'batch', 1: 'choice', 2: 'sequence'}
						else:
									__A=	{0: 'batch', 1: 'sequence'}
						return OrderedDict(
						    [
						        ('input_ids', dynamic_axis),
						        ('attention_mask', dynamic_axis),
						    ]	)







'''simple docstring'''





# Lint as: python3
import itertools
import os
import re


UpperCAmelCase__  			=				re.compile(r'''([A-Z]+)([A-Z][a-z])''')
UpperCAmelCase__  			=				re.compile(r'''([a-z\d])([A-Z])''')

UpperCAmelCase__  			=				re.compile(r'''(?<!_)_(?!_)''')
UpperCAmelCase__  			=				re.compile(r'''(_{2,})''')

UpperCAmelCase__  			=				r'''^\w+(\.\w+)*$'''

UpperCAmelCase__  			=				r'''<>:/\|?*'''






def 						UpperCAmelCase__(				_SCREAMING_SNAKE_CASE					: Union[str, Any] ):






			"""simple docstring"""




			__A=	_uppercase_uppercase_re.sub(r'\1_\2',_SCREAMING_SNAKE_CASE )
			__A=	_lowercase_uppercase_re.sub(r'\1_\2',_SCREAMING_SNAKE_CASE )
			return name.lower()






def 						UpperCAmelCase__(				_SCREAMING_SNAKE_CASE					: int ):






			"""simple docstring"""




			__A=	_single_underscore_re.split(_SCREAMING_SNAKE_CASE )
			__A=	[_multiple_underscores_re.split(_SCREAMING_SNAKE_CASE ) for n in name]
			return "".join(n.capitalize() for n in itertools.chain.from_iterable(_SCREAMING_SNAKE_CASE ) if n != '' )






def 						UpperCAmelCase__(				_SCREAMING_SNAKE_CASE					: str ):






			"""simple docstring"""




			if os.path.basename(_SCREAMING_SNAKE_CASE ) != name:
						raise ValueError(f"""Should be a dataset name, not a path: {name}""" )
			return camelcase_to_snakecase(_SCREAMING_SNAKE_CASE )






def 						UpperCAmelCase__(				_SCREAMING_SNAKE_CASE					: Union[str, Any],_SCREAMING_SNAKE_CASE					: str ):






			"""simple docstring"""




			if os.path.basename(_SCREAMING_SNAKE_CASE ) != name:
						raise ValueError(f"""Should be a dataset name, not a path: {name}""" )
			if not re.match(_split_re,_SCREAMING_SNAKE_CASE ):
						raise ValueError(f"""Split name should match '{_split_re}'' but got '{split}'.""" )
			return f"""{filename_prefix_for_name(_SCREAMING_SNAKE_CASE )}-{split}"""






def 						UpperCAmelCase__(				_SCREAMING_SNAKE_CASE					: Union[str, Any],_SCREAMING_SNAKE_CASE					: Optional[Any],_SCREAMING_SNAKE_CASE					: Optional[int],_SCREAMING_SNAKE_CASE					: Union[str, Any]=None ):






			"""simple docstring"""




			__A=	filename_prefix_for_split(_SCREAMING_SNAKE_CASE,_SCREAMING_SNAKE_CASE )
			if filetype_suffix:
						prefix += f""".{filetype_suffix}"""
			__A=	os.path.join(_SCREAMING_SNAKE_CASE,_SCREAMING_SNAKE_CASE )
			return f"""{filepath}*"""






def 						UpperCAmelCase__(				_SCREAMING_SNAKE_CASE					: int,_SCREAMING_SNAKE_CASE					: List[str],_SCREAMING_SNAKE_CASE					: int,_SCREAMING_SNAKE_CASE					: List[Any]=None,_SCREAMING_SNAKE_CASE					: List[str]=None ):






			"""simple docstring"""




			__A=	filename_prefix_for_split(_SCREAMING_SNAKE_CASE,_SCREAMING_SNAKE_CASE )
			__A=	os.path.join(_SCREAMING_SNAKE_CASE,_SCREAMING_SNAKE_CASE )

			if shard_lengths:
						__A=	len(_SCREAMING_SNAKE_CASE )
						__A=	[f"""{prefix}-{shard_id:05d}-of-{num_shards:05d}""" for shard_id in range(_SCREAMING_SNAKE_CASE )]
						if filetype_suffix:
									__A=	[filename + f""".{filetype_suffix}""" for filename in filenames]
						return filenames
			else:
						__A=	prefix
						if filetype_suffix:
									filename += f""".{filetype_suffix}"""
						return [filename]



