code: 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    __lowerCAmelCase  (       a_						):
  
  
  
   '''simple docstring'''
   _lowerCamelCase						:	Tuple				    = 'swiftformer'
  
   def __init__(			self       :		List[Any]       ,			lowerCamelCase       :		Union[str, Any]=3       ,			lowerCamelCase       :		str=[3, 3, 6, 4]       ,			lowerCamelCase       :		Union[str, Any]=[4_8, 5_6, 1_1_2, 2_2_0]       ,			lowerCamelCase       :		List[str]=4       ,			lowerCamelCase       :		Optional[Any]=[True, True, True, True]       ,			lowerCamelCase       :		Union[str, Any]="gelu"       ,			lowerCamelCase       :		Union[str, Any]=3       ,			lowerCamelCase       :		Tuple=2       ,			lowerCamelCase       :		Dict=1       ,			lowerCamelCase       :		str=0.0       ,			lowerCamelCase       :		List[Any]=True       ,			lowerCamelCase       :		Optional[Any]=1e-5       ,			lowerCamelCase       :		List[Any]=1e-5       ,			**lowerCamelCase       :		List[str]       ,			) ->							Tuple:
          super().__init__(**lowerCamelCase  )
          _UpperCamelCase     :       Optional[int]  						=  num_channels
          _UpperCamelCase     :       Any  						=  depths
          _UpperCamelCase     :       List[Any]  						=  embed_dims
          _UpperCamelCase     :       Union[str, Any]  						=  mlp_ratio
          _UpperCamelCase     :       List[Any]  						=  downsamples
          _UpperCamelCase     :       Any  						=  hidden_act
          _UpperCamelCase     :       Optional[Any]  						=  down_patch_size
          _UpperCamelCase     :       Optional[int]  						=  down_stride
          _UpperCamelCase     :       Union[str, Any]  						=  down_pad
          _UpperCamelCase     :       Tuple  						=  drop_path_rate
          _UpperCamelCase     :       Union[str, Any]  						=  use_layer_scale
          _UpperCamelCase     :       List[str]  						=  layer_scale_init_value
          _UpperCamelCase     :       List[Any]  						=  batch_norm_eps
  
  
  
  
  
  
  
  class    __lowerCAmelCase  (       a_						):
  
  
  
   '''simple docstring'''
   _lowerCamelCase						:	Union[str, Any]				    = version.parse('1.11'						)
  
   @property
   def 					__snake_case					(			self       :		str  ) ->							Mapping[str, Mapping[int, str]]:
          return OrderedDict(
              [
                  ('pixel_values', {0: 'batch', 1: 'num_channels', 2: 'height', 3: 'width'}),
              ]  )
  
   @property
   def 					__snake_case					(			self       :		Optional[Any]  ) ->							float:
          return 1e-4
  
  
  
  
style_context: from .chunk_utils import chunk_layer
  from .data_transforms import make_atomaa_masks
  from .feats import atomaa_to_atomaa, frames_and_literature_positions_to_atomaa_pos, torsion_angles_to_frames
  from .loss import compute_predicted_aligned_error, compute_tm
  from .protein import Protein as OFProtein
  from .protein import to_pdb
  from .rigid_utils import Rigid, Rotation
  from .tensor_utils import dict_multimap, flatten_final_dims, permute_final_dims
  
  
  
  