





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


snake_case_     :       Optional[Any]						=  logging.get_logger(__name__)

snake_case_     :       Tuple						=  {
    "facebook/levit-128S": "https://huggingface.co/facebook/levit-128S/resolve/main/config.json",
    # See all LeViT models at https://huggingface.co/models?filter=levit
}


class 					__a      (lowerCamelCase     ):
							__a					:   Tuple									=       "levit"

							def __init__(							self							:     List[str]       ,     __magic_name__							:     Optional[Any]=2_24       ,     __magic_name__							:     Tuple=3       ,     __magic_name__							:     List[str]=3       ,     __magic_name__							:     Optional[int]=2       ,     __magic_name__							:     Optional[Any]=1       ,     __magic_name__							:     Optional[Any]=16       ,     __magic_name__							:     str=[1_28, 2_56, 3_84]       ,     __magic_name__							:     Union[str, Any]=[4, 8, 12]       ,     __magic_name__							:     int=[4, 4, 4]       ,     __magic_name__							:     List[Any]=[16, 16, 16]       ,     __magic_name__							:     Dict=0       ,     __magic_name__							:     Union[str, Any]=[2, 2, 2]       ,     __magic_name__							:     int=[2, 2, 2]       ,     __magic_name__							:     Any=0.0_2       ,     **__magic_name__							:     Union[str, Any]       ,     )					->       Dict:







													"""simple docstring"""
													super().__init__(**__magic_name__   )
													UpperCAmelCase_			:      Tuple     						=  image_size
													UpperCAmelCase_			:      str     						=  num_channels
													UpperCAmelCase_			:      Tuple     						=  kernel_size
													UpperCAmelCase_			:      int     						=  stride
													UpperCAmelCase_			:      List[Any]     						=  padding
													UpperCAmelCase_			:      int     						=  hidden_sizes
													UpperCAmelCase_			:      int     						=  num_attention_heads
													UpperCAmelCase_			:      Union[str, Any]     						=  depths
													UpperCAmelCase_			:      List[Any]     						=  key_dim
													UpperCAmelCase_			:      Tuple     						=  drop_path_rate
													UpperCAmelCase_			:      Optional[Any]     						=  patch_size
													UpperCAmelCase_			:      Tuple     						=  attention_ratio
													UpperCAmelCase_			:      Union[str, Any]     						=  mlp_ratio
													UpperCAmelCase_			:      Dict     						=  initializer_range
													UpperCAmelCase_			:      Tuple     						=  [
													    ['''Subsample''', key_dim[0], hidden_sizes[0] // key_dim[0], 4, 2, 2],
													    ['''Subsample''', key_dim[0], hidden_sizes[1] // key_dim[0], 4, 2, 2],
													]





class 					__a      (lowerCamelCase     ):
							__a					:   Optional[Any]									=       version.parse("1.11"     )

							@property
							def     UpperCAmelCase__	(							self							:     Optional[int]   )					->       Mapping[str, Mapping[int, str]]:







													"""simple docstring"""
													return OrderedDict(
													    [
													        ('''pixel_values''', {0: '''batch''', 1: '''num_channels''', 2: '''height''', 3: '''width'''}),
													    ]   )



							@property
							def     UpperCAmelCase__	(							self							:     Any   )					->       float:







													"""simple docstring"""
													return 1E-4












'''simple docstring'''







snake_case_     :       int						=  {
    "Pillow": "Pillow",
    "accelerate": "accelerate>=0.11.0",
    "compel": "compel==0.1.8",
    "black": "black~=23.1",
    "datasets": "datasets",
    "filelock": "filelock",
    "flax": "flax>=0.4.1",
    "hf-doc-builder": "hf-doc-builder>=0.3.0",
    "huggingface-hub": "huggingface-hub>=0.13.2",
    "requests-mock": "requests-mock==1.10.0",
    "importlib_metadata": "importlib_metadata",
    "invisible-watermark": "invisible-watermark",
    "isort": "isort>=5.5.4",
    "jax": "jax>=0.2.8,!=0.3.2",
    "jaxlib": "jaxlib>=0.1.65",
    "Jinja2": "Jinja2",
    "k-diffusion": "k-diffusion>=0.0.12",
    "torchsde": "torchsde",
    "note_seq": "note_seq",
    "librosa": "librosa",
    "numpy": "numpy",
    "omegaconf": "omegaconf",
    "parameterized": "parameterized",
    "protobuf": "protobuf>=3.20.3,<4",
    "pytest": "pytest",
    "pytest-timeout": "pytest-timeout",
    "pytest-xdist": "pytest-xdist",
    "ruff": "ruff>=0.0.241",
    "safetensors": "safetensors",
    "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
    "scipy": "scipy",
    "onnx": "onnx",
    "regex": "regex!=2019.12.17",
    "requests": "requests",
    "tensorboard": "tensorboard",
    "torch": "torch>=1.4",
    "torchvision": "torchvision",
    "transformers": "transformers>=4.25.1",
    "urllib3": "urllib3<=2.0.0",
}





