

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


A_	= logging.get_logger(__name__)

A_	= {
    "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  UpperCAmelCase (							A__    ):

		'''simple docstring'''






		SCREAMING_SNAKE_CASE_           =					"""levit"""
		def __init__(					self   ,			SCREAMING_SNAKE_CASE_=224   ,			SCREAMING_SNAKE_CASE_=3   ,			SCREAMING_SNAKE_CASE_=3   ,			SCREAMING_SNAKE_CASE_=2   ,			SCREAMING_SNAKE_CASE_=1   ,			SCREAMING_SNAKE_CASE_=16   ,			SCREAMING_SNAKE_CASE_=[128, 256, 384]   ,			SCREAMING_SNAKE_CASE_=[4, 8, 12]   ,			SCREAMING_SNAKE_CASE_=[4, 4, 4]   ,			SCREAMING_SNAKE_CASE_=[16, 16, 16]   ,			SCREAMING_SNAKE_CASE_=0   ,			SCREAMING_SNAKE_CASE_=[2, 2, 2]   ,			SCREAMING_SNAKE_CASE_=[2, 2, 2]   ,			SCREAMING_SNAKE_CASE_=0.02   ,			**SCREAMING_SNAKE_CASE_   ,			)  ->     List[str]:






							'''simple docstring'''



							super().__init__(**_UpperCamelCase		)
							lowerCamelCase_  				=      image_size
							lowerCamelCase_  				=      num_channels
							lowerCamelCase_  				=      kernel_size
							lowerCamelCase_  				=      stride
							lowerCamelCase_  				=      padding
							lowerCamelCase_  				=      hidden_sizes
							lowerCamelCase_  				=      num_attention_heads
							lowerCamelCase_  				=      depths
							lowerCamelCase_  				=      key_dim
							lowerCamelCase_  				=      drop_path_rate
							lowerCamelCase_  				=      patch_size
							lowerCamelCase_  				=      attention_ratio
							lowerCamelCase_  				=      mlp_ratio
							lowerCamelCase_  				=      initializer_range
							lowerCamelCase_  				=      [
							    ['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  UpperCAmelCase (							A__    ):

		'''simple docstring'''






		SCREAMING_SNAKE_CASE_           =					version.parse('1.11'    )
		@property
		def 			UpperCamelCase(					self		)  ->     Optional[Any]:






							'''simple docstring'''



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






		@property
		def 			UpperCamelCase(					self		)  ->     Dict:






							'''simple docstring'''



							return 1E-4










from math import sqrt
def      A__       (    SCREAMING_SNAKE_CASE_   :      int						)  ->  bool:






   """simple docstring"""







   if 1 < number < 4:
      # 2 and 3 are primes
      return True
   elif number < 2 or number % 2 == 0 or number % 3 == 0:
      # Negatives, 0, 1, all even numbers, all multiples of 3 are not primes
      return False

   # All primes number are in format of 6k +/- 1
   for i in range(5				,	int(sqrt(SCREAMING_SNAKE_CASE_						) + 1						)				,	6						):
      if number % i == 0 or number % (i + 2) == 0:
         return False
   return True
def      A__       (    SCREAMING_SNAKE_CASE_   :      int = 1_00_01						)  ->  int:






   """simple docstring"""







   _UpperCAmelCase           =   0
   _UpperCAmelCase           =   1
   while count != nth and number < 3:
      number += 1
      if is_prime(SCREAMING_SNAKE_CASE_						):
         count += 1
   while count != nth:
      number += 2
      if is_prime(SCREAMING_SNAKE_CASE_						):
         count += 1
   return number


if __name__ == "__main__":
   print(f'''{solution() = }''')