


"""simple docstring"""



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    __a						( __snake_case							):
				lowerCamelCase     :       Any			       ='distilbert'
				lowerCamelCase     :       Optional[int]			       ={
				    'hidden_size': 'dim',
				    'num_attention_heads': 'n_heads',
				    'num_hidden_layers': 'n_layers',
				}







				def __init__( self      ,	UpperCAmelCase=3_0522      ,	UpperCAmelCase=512      ,	UpperCAmelCase=False      ,	UpperCAmelCase=6      ,	UpperCAmelCase=12      ,	UpperCAmelCase=768      ,	UpperCAmelCase=4 * 768      ,	UpperCAmelCase=0.1      ,	UpperCAmelCase=0.1      ,	UpperCAmelCase="gelu"      ,	UpperCAmelCase=0.0_2      ,	UpperCAmelCase=0.1      ,	UpperCAmelCase=0.2      ,	UpperCAmelCase=0      ,	**UpperCAmelCase      ,	):

					'''simple docstring'''






					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__(**UpperCAmelCase      ,	pad_token_id=UpperCAmelCase   )







class    __a						( __snake_case							):







				@property
				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					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"""



from diffusers.utils.testing_utils import require_onnxruntime







@require_onnxruntime
class    __a						:
				pass