
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    			=					{
    'MBZUAI/swiftformer-xs': 'https://huggingface.co/MBZUAI/swiftformer-xs/resolve/main/config.json',
}

class 		__magic_name__				( UpperCAmelCase_					):


  """simple docstring"""





  _UpperCamelCase						 = "swiftformer"


  def __init__(					self	,		a__=3	,		a__=[3, 3, 6, 4]	,		a__=[48, 56, 1_12, 2_20]	,		a__=4	,		a__=[True, True, True, True]	,		a__="gelu"	,		a__=3	,		a__=2	,		a__=1	,		a__=0.0	,		a__=True	,		a__=1E-5	,		a__=1E-5	,		**a__	,		):
    super().__init__(**__UpperCamelCase		)
    _lowerCamelCase      =						num_channels
    _lowerCamelCase      =						depths
    _lowerCamelCase      =						embed_dims
    _lowerCamelCase      =						mlp_ratio
    _lowerCamelCase      =						downsamples
    _lowerCamelCase      =						hidden_act
    _lowerCamelCase      =						down_patch_size
    _lowerCamelCase      =						down_stride
    _lowerCamelCase      =						down_pad
    _lowerCamelCase      =						drop_path_rate
    _lowerCamelCase      =						use_layer_scale
    _lowerCamelCase      =						layer_scale_init_value
    _lowerCamelCase      =						batch_norm_eps







class 		__magic_name__				( UpperCAmelCase_					):


  """simple docstring"""





  _UpperCamelCase						 = version.parse("1.11"					)


  @property
  def     _UpperCAmelCase  (					self		):
    return OrderedDict(
        [
            ('''pixel_values''', {0: '''batch''', 1: '''num_channels''', 2: '''height''', 3: '''width'''}),
        ]		)


  @property
  def     _UpperCAmelCase  (					self		):
    return 1E-4










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


_lowerCamelCase				:		List[str]				      =    logging.get_logger(__name__)

_lowerCamelCase				:		Optional[Any]				      =    {
    'MBZUAI/swiftformer-xs': 'https://huggingface.co/MBZUAI/swiftformer-xs/resolve/main/config.json',
}





class 	lowerCamelCase_				(			UpperCAmelCase_				):




					'''simple docstring'''






					lowercase__       =					"""swiftformer"""






					def __init__(       self,    __UpperCamelCase=3,    __UpperCamelCase=[3, 3, 6, 4],    __UpperCamelCase=[48, 56, 112, 220],    __UpperCamelCase=4,    __UpperCamelCase=[True, True, True, True],    __UpperCamelCase="gelu",    __UpperCamelCase=3,    __UpperCamelCase=2,    __UpperCamelCase=1,    __UpperCamelCase=0.0,    __UpperCamelCase=True,    __UpperCamelCase=1E-5,    __UpperCamelCase=1E-5,    **__UpperCamelCase,    ):
							super().__init__(**__UpperCamelCase					)
							_snake_case      			=			num_channels
							_snake_case      			=			depths
							_snake_case      			=			embed_dims
							_snake_case      			=			mlp_ratio
							_snake_case      			=			downsamples
							_snake_case      			=			hidden_act
							_snake_case      			=			down_patch_size
							_snake_case      			=			down_stride
							_snake_case      			=			down_pad
							_snake_case      			=			drop_path_rate
							_snake_case      			=			use_layer_scale
							_snake_case      			=			layer_scale_init_value
							_snake_case      			=			batch_norm_eps





class 	lowerCamelCase_				(			UpperCAmelCase_				):




					'''simple docstring'''






					lowercase__       =					version.parse("""1.11"""				)






					@property
					def 							__SCREAMING_SNAKE_CASE      (       self					):
							return OrderedDict(
							    [
							        ("pixel_values", {0: "batch", 1: "num_channels", 2: "height", 3: "width"}),
							    ]					)






					@property
					def 							__SCREAMING_SNAKE_CASE      (       self					):
							return 1E-4





