

"""simple docstring"""


from collections import OrderedDict
from typing import Mapping

from packaging import version

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


UpperCAmelCase													=logging.get_logger(__name__)

UpperCAmelCase													={
    "facebook/deit-base-distilled-patch16-224": (
        "https://huggingface.co/facebook/deit-base-patch16-224/resolve/main/config.json"
    ),
    # See all DeiT models at https://huggingface.co/models?filter=deit
}





class lowerCamelCase__   (				SCREAMING_SNAKE_CASE    ):



    '''simple docstring'''




    _lowerCamelCase					=   '''deit'''




    def __init__( self ,lowerCamelCase_=7_6_8 ,lowerCamelCase_=1_2 ,lowerCamelCase_=1_2 ,lowerCamelCase_=3_0_7_2 ,lowerCamelCase_="gelu" ,lowerCamelCase_=0.0 ,lowerCamelCase_=0.0 ,lowerCamelCase_=0.02 ,lowerCamelCase_=1E-12 ,lowerCamelCase_=2_2_4 ,lowerCamelCase_=1_6 ,lowerCamelCase_=3 ,lowerCamelCase_=True ,lowerCamelCase_=1_6 ,**lowerCamelCase_ ,)    ->      Dict:
          super().__init__(**lowerCamelCase_    )

          A				  =		hidden_size
          A				  =		num_hidden_layers
          A				  =		num_attention_heads
          A				  =		intermediate_size
          A				  =		hidden_act
          A				  =		hidden_dropout_prob
          A				  =		attention_probs_dropout_prob
          A				  =		initializer_range
          A				  =		layer_norm_eps
          A				  =		image_size
          A				  =		patch_size
          A				  =		num_channels
          A				  =		qkv_bias
          A				  =		encoder_stride





class lowerCamelCase__   (				SCREAMING_SNAKE_CASE    ):



    '''simple docstring'''




    _lowerCamelCase					=   version.parse('''1.11'''    )




    @property
    def UpperCamelCase__		( self    )    ->      Mapping[str, Mapping[int, str]]:
          return OrderedDict(
              [
                  ("""pixel_values""", {0: """batch""", 1: """num_channels""", 2: """height""", 3: """width"""}),
              ]    )




    @property
    def UpperCamelCase__		( self    )    ->      float:
          return 1E-4







"""simple docstring"""


from dataclasses import dataclass, field
from typing import ClassVar, Dict

from ..features import Features, Value
from .base import TaskTemplate





@dataclass(frozen=SCREAMING_SNAKE_CASE    )
class lowerCamelCase__   (				SCREAMING_SNAKE_CASE    ):



    '''simple docstring'''




    _lowerCamelCase					=   field(default='''summarization'''     ,   metadata={'''include_in_asdict_even_if_is_default''': True}    )
    _lowerCamelCase					=   Features({'''text''': Value('''string'''    )}    )
    _lowerCamelCase					=   Features({'''summary''': Value('''string'''    )}    )
    _lowerCamelCase					=   "text"
    _lowerCamelCase					=   "summary"




    @property
    def UpperCamelCase__		( self    )    ->      Dict[str, str]:
          return {self.text_column: "text", self.summary_column: "summary"}




