from collections import OrderedDict
from typing import Mapping

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


lowercase_     		= logging.get_logger(__name__)

lowercase_     		= {
    '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   SCREAMING_SNAKE_CASE			(						__lowercase):







		"""simple docstring"""
		lowercase   :      Optional[int]       =							'distilbert'
		lowercase   :      Union[str, Any]       =							{
		    'hidden_size': 'dim',
		    'num_attention_heads': 'n_heads',
		    'num_hidden_layers': 'n_layers',
		}






		def __init__(   self		,						__A=3_0522		,						__A=512		,						__A=False		,						__A=6		,						__A=12		,						__A=768		,						__A=4 * 768		,						__A=0.1		,						__A=0.1		,						__A="gelu"		,						__A=0.02		,						__A=0.1		,						__A=0.2		,						__A=0		,						**__A		,						) ->     Tuple:
								_lowerCAmelCase										=vocab_size
								_lowerCAmelCase										=max_position_embeddings
								_lowerCAmelCase										=sinusoidal_pos_embds
								_lowerCAmelCase										=n_layers
								_lowerCAmelCase										=n_heads
								_lowerCAmelCase										=dim
								_lowerCAmelCase										=hidden_dim
								_lowerCAmelCase										=dropout
								_lowerCAmelCase										=attention_dropout
								_lowerCAmelCase										=activation
								_lowerCAmelCase										=initializer_range
								_lowerCAmelCase										=qa_dropout
								_lowerCAmelCase										=seq_classif_dropout
								super().__init__(**__A		,						pad_token_id=__A				)

class   SCREAMING_SNAKE_CASE			(						__lowercase):







		"""simple docstring"""






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






'''simple docstring'''

import json
from typing import List, Optional, Tuple

from tokenizers import normalizers

from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_convbert import ConvBertTokenizer


lowercase_     		= logging.get_logger(__name__)

lowercase_     		= {'''vocab_file''': '''vocab.txt'''}

lowercase_     		= {
    '''vocab_file''': {
        '''YituTech/conv-bert-base''': '''https://huggingface.co/YituTech/conv-bert-base/resolve/main/vocab.txt''',
        '''YituTech/conv-bert-medium-small''': (
            '''https://huggingface.co/YituTech/conv-bert-medium-small/resolve/main/vocab.txt'''
        ),
        '''YituTech/conv-bert-small''': '''https://huggingface.co/YituTech/conv-bert-small/resolve/main/vocab.txt''',
    }
}

lowercase_     		= {
    '''YituTech/conv-bert-base''': 512,
    '''YituTech/conv-bert-medium-small''': 512,
    '''YituTech/conv-bert-small''': 512,
}


lowercase_     		= {
    '''YituTech/conv-bert-base''': {'''do_lower_case''': True},
    '''YituTech/conv-bert-medium-small''': {'''do_lower_case''': True},
    '''YituTech/conv-bert-small''': {'''do_lower_case''': True},
}
class   SCREAMING_SNAKE_CASE			(						__lowercase):







		"""simple docstring"""
		lowercase   :      Union[str, Any]       =							VOCAB_FILES_NAMES
		lowercase   :      Tuple       =							PRETRAINED_VOCAB_FILES_MAP
		lowercase   :      Optional[int]       =							PRETRAINED_INIT_CONFIGURATION
		lowercase   :      int       =							PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
		lowercase   :      List[str]       =							ConvBertTokenizer






		def __init__(   self		,						__A=None		,						__A=None		,						__A=True		,						__A="[UNK]"		,						__A="[SEP]"		,						__A="[PAD]"		,						__A="[CLS]"		,						__A="[MASK]"		,						__A=True		,						__A=None		,						**__A		,						) ->     Union[str, Any]:
								super().__init__(
								    __A		,						tokenizer_file=__A		,						do_lower_case=__A		,						unk_token=__A		,						sep_token=__A		,						pad_token=__A		,						cls_token=__A		,						mask_token=__A		,						tokenize_chinese_chars=__A		,						strip_accents=__A		,						**__A		,						)

								_lowerCAmelCase										=json.loads(self.backend_tokenizer.normalizer.__getstate__()				)
								if (
								    normalizer_state.get('lowercase'		,						__A				) != do_lower_case
								    or normalizer_state.get('strip_accents'		,						__A				) != strip_accents
								    or normalizer_state.get('handle_chinese_chars'		,						__A				) != tokenize_chinese_chars
								):
														_lowerCAmelCase										=getattr(__A		,						normalizer_state.pop('type'				)				)
														_lowerCAmelCase										=do_lower_case
														_lowerCAmelCase										=strip_accents
														_lowerCAmelCase										=tokenize_chinese_chars
														_lowerCAmelCase										=normalizer_class(**__A				)

								_lowerCAmelCase										=do_lower_case






		def 		UpperCamelCase__       (   self		,						__A		,						__A=None				) ->     int:
								_lowerCAmelCase										=[self.cls_token_id] + token_ids_a + [self.sep_token_id]

								if token_ids_a:
														output += token_ids_a + [self.sep_token_id]

								return output






		def 		UpperCamelCase__       (   self		,						__A		,						__A = None				) ->     List[int]:
								_lowerCAmelCase										=[self.sep_token_id]
								_lowerCAmelCase										=[self.cls_token_id]
								if token_ids_a is None:
														return len(cls + token_ids_a + sep				) * [0]
								return len(cls + token_ids_a + sep				) * [0] + len(token_ids_a + sep				) * [1]






		def 		UpperCamelCase__       (   self		,						__A		,						__A = None				) ->     Tuple[str]:
								_lowerCAmelCase										=self._tokenizer.model.save(__A		,						name=__A				)
								return tuple(__A				)





