
from ...configuration_utils import PretrainedConfig
from ...utils import logging


_snake_case      :							str     	=    logging.get_logger(__name__)

_snake_case      :							int     	=    {
    "facebook/s2t-small-librispeech-asr": (
        "https://huggingface.co/facebook/s2t-small-librispeech-asr/resolve/main/config.json"
    ),
    # See all Speech2Text models at https://huggingface.co/models?filter=speech_to_text
}





class   UpperCamelCase_      (      __a		):







						'''simple docstring'''



						UpperCamelCase      :  List[Any]				   =					'''speech_to_text'''
						UpperCamelCase      :  Dict				   =					['''past_key_values''']
						UpperCamelCase      :  List[str]				   =					{'''num_attention_heads''': '''encoder_attention_heads''', '''hidden_size''': '''d_model'''}





						def __init__(			self							:str		,    lowerCAmelCase__							:int=10000		,    lowerCAmelCase__							:str=12		,    lowerCAmelCase__							:Optional[Any]=2048		,    lowerCAmelCase__							:int=4		,    lowerCAmelCase__							:Optional[Any]=6		,    lowerCAmelCase__							:Union[str, Any]=2048		,    lowerCAmelCase__							:Optional[Any]=4		,    lowerCAmelCase__							:int=0.0		,    lowerCAmelCase__							:List[str]=0.0		,    lowerCAmelCase__							:List[str]=True		,    lowerCAmelCase__							:Dict=True		,    lowerCAmelCase__							:Optional[int]="relu"		,    lowerCAmelCase__							:int=256		,    lowerCAmelCase__							:Union[str, Any]=0.1		,    lowerCAmelCase__							:List[Any]=0.0		,    lowerCAmelCase__							:List[str]=0.0		,    lowerCAmelCase__							:Any=0.02		,    lowerCAmelCase__							:List[Any]=2		,    lowerCAmelCase__							:int=True		,    lowerCAmelCase__							:Tuple=1		,    lowerCAmelCase__							:List[Any]=0		,    lowerCAmelCase__							:Union[str, Any]=2		,    lowerCAmelCase__							:str=6000		,    lowerCAmelCase__							:Optional[Any]=1024		,    lowerCAmelCase__							:Optional[Any]=2		,    lowerCAmelCase__							:List[Any]=(5, 5)		,    lowerCAmelCase__							:Union[str, Any]=1024		,    lowerCAmelCase__							:int=80		,    lowerCAmelCase__							:Optional[int]=1		,    **lowerCAmelCase__							:Union[str, Any]		,    )  ->Any:
										lowercase         =      vocab_size
										lowercase         =      d_model
										lowercase         =      encoder_ffn_dim
										lowercase         =      encoder_layers
										lowercase         =      encoder_attention_heads
										lowercase         =      decoder_ffn_dim
										lowercase         =      decoder_layers
										lowercase         =      decoder_attention_heads
										lowercase         =      dropout
										lowercase         =      attention_dropout
										lowercase         =      activation_dropout
										lowercase         =      activation_function
										lowercase         =      init_std
										lowercase         =      encoder_layerdrop
										lowercase         =      decoder_layerdrop
										lowercase         =      use_cache
										lowercase         =      encoder_layers
										lowercase         =      scale_embedding  # scale factor will be sqrt(d_model) if True
										lowercase         =      max_source_positions
										lowercase         =      max_target_positions
										lowercase         =      num_conv_layers
										lowercase         =      list(lowerCAmelCase__     )
										lowercase         =      conv_channels
										lowercase         =      input_feat_per_channel
										lowercase         =      input_channels

										if len(self.conv_kernel_sizes     ) != self.num_conv_layers:
														raise ValueError(
														    "Configuration for convolutional module is incorrect. "
														    "It is required that `len(config.conv_kernel_sizes)` == `config.num_conv_layers` "
														    F'''but is `len(config.conv_kernel_sizes) = {len(self.conv_kernel_sizes     )}`, '''
														    F'''`config.num_conv_layers = {self.num_conv_layers}`.'''     )

										super().__init__(
										    pad_token_id=lowerCAmelCase__		,    bos_token_id=lowerCAmelCase__		,    eos_token_id=lowerCAmelCase__		,    is_encoder_decoder=lowerCAmelCase__		,    decoder_start_token_id=lowerCAmelCase__		,    **lowerCAmelCase__		,    )


def  __snake_case							(	__magic_name__		,    __magic_name__ = False						):







				'''simple docstring'''


				if n == 2:
								return True
				if not n % 2 or n < 2:
								return False
				if n > 5 and n % 10 not in (1, 3, 7, 9):  # can quickly check last digit
								return False
				if n > 3317044064679887385961981 and not allow_probable:
								raise ValueError(
								    "Warning: upper bound of deterministic test is exceeded. "
								    "Pass allow_probable=True to allow probabilistic test. "
								    "A return value of True indicates a probable prime."						)
				# array bounds provided by analysis
				lowercase         =      [
				    2047,
				    1373653,
				    25326001,
				    3215031751,
				    2152302898747,
				    3474749660383,
				    341550071728321,
				    1,
				    3825123056546413051,
				    1,
				    1,
				    318665857834031151167461,
				    3317044064679887385961981,
				]

				lowercase         =      [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41]
				for idx, _p in enumerate(__magic_name__		,    1						):
								if n < _p:
												# then we have our last prime to check
												lowercase         =      primes[:idx]
												break
				lowercase     ,	lowercase         =      n - 1, 0
				# break up n -1 into a power of 2 (s) and
				# remaining odd component
				# essentially, solve for d * 2 ** s == n - 1
				while d % 2 == 0:
								d //= 2
								s += 1
				for prime in plist:
								lowercase         =      False
								for r in range(__magic_name__						):
												lowercase         =      pow(__magic_name__		,    d * 2**r		,    __magic_name__						)
												# see article for analysis explanation for m
												if (r == 0 and m == 1) or ((m + 1) % n == 0):
																lowercase         =      True
																# this loop will not determine compositeness
																break
								if pr:
												continue
								# if pr is False, then the above loop never evaluated to true,
								# and the n MUST be composite
								return False
				return True

def  __snake_case							(	):







				'''simple docstring'''


				assert not miller_rabin(561						)
				assert miller_rabin(563						)
				# 2047

				assert not miller_rabin(838201						)
				assert miller_rabin(838207						)
				# 1_373_653

				assert not miller_rabin(17316001						)
				assert miller_rabin(17316017						)
				# 25_326_001

				assert not miller_rabin(3078386641						)
				assert miller_rabin(3078386653						)
				# 3_215_031_751

				assert not miller_rabin(1713045574801						)
				assert miller_rabin(1713045574819						)
				# 2_152_302_898_747

				assert not miller_rabin(2779799728307						)
				assert miller_rabin(2779799728327						)
				# 3_474_749_660_383

				assert not miller_rabin(113850023909441						)
				assert miller_rabin(113850023909527						)
				# 341_550_071_728_321

				assert not miller_rabin(1275041018848804351						)
				assert miller_rabin(1275041018848804391						)
				# 3_825_123_056_546_413_051

				assert not miller_rabin(79666464458507787791867						)
				assert miller_rabin(79666464458507787791951						)
				# 318_665_857_834_031_151_167_461

				assert not miller_rabin(552840677446647897660333						)
				assert miller_rabin(552840677446647897660359						)
				# 3_317_044_064_679_887_385_961_981
				# upper limit for probabilistic test


if __name__ == "__main__":
						test_miller_rabin()
