





"""simple docstring"""





from collections import OrderedDict
from typing import Mapping

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


SCREAMING_SNAKE_CASE__					=   logging.get_logger(__name__)

SCREAMING_SNAKE_CASE__					=   {
    '''google/bigbird-roberta-base''': '''https://huggingface.co/google/bigbird-roberta-base/resolve/main/config.json''',
    '''google/bigbird-roberta-large''': '''https://huggingface.co/google/bigbird-roberta-large/resolve/main/config.json''',
    '''google/bigbird-base-trivia-itc''': '''https://huggingface.co/google/bigbird-base-trivia-itc/resolve/main/config.json''',
    # See all BigBird models at https://huggingface.co/models?filter=big_bird
}

class 		lowercase					(					__a					):
      _SCREAMING_SNAKE_CASE					=      '''big_bird'''





      def __init__(       self      ,						lowercase=50_358      ,						lowercase=768      ,						lowercase=12      ,						lowercase=12      ,						lowercase=3_072      ,						lowercase="gelu_new"      ,						lowercase=0.1      ,						lowercase=0.1      ,						lowercase=4_096      ,						lowercase=2      ,						lowercase=0.02      ,						lowercase=1e-12      ,						lowercase=True      ,						lowercase=0      ,						lowercase=1      ,						lowercase=2      ,						lowercase=66      ,						lowercase="block_sparse"      ,						lowercase=True      ,						lowercase=False      ,						lowercase=64      ,						lowercase=3      ,						lowercase=None      ,						**lowercase      ,						)		->     Optional[Any]:
            super().__init__(
                pad_token_id=__a      ,						bos_token_id=__a      ,						eos_token_id=__a      ,						sep_token_id=__a      ,						**__a      ,						)

            lowerCAmelCase  						=       vocab_size
            lowerCAmelCase  						=       max_position_embeddings
            lowerCAmelCase  						=       hidden_size
            lowerCAmelCase  						=       num_hidden_layers
            lowerCAmelCase  						=       num_attention_heads
            lowerCAmelCase  						=       intermediate_size
            lowerCAmelCase  						=       hidden_act
            lowerCAmelCase  						=       hidden_dropout_prob
            lowerCAmelCase  						=       attention_probs_dropout_prob
            lowerCAmelCase  						=       initializer_range
            lowerCAmelCase  						=       type_vocab_size
            lowerCAmelCase  						=       layer_norm_eps
            lowerCAmelCase  						=       use_cache

            lowerCAmelCase  						=       rescale_embeddings
            lowerCAmelCase  						=       attention_type
            lowerCAmelCase  						=       use_bias
            lowerCAmelCase  						=       block_size
            lowerCAmelCase  						=       num_random_blocks
            lowerCAmelCase  						=       classifier_dropout







class 		lowercase					(					__a					):





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

            if self.task == "multiple-choice":
                  lowerCAmelCase  						=       {0: """batch""", 1: """choice""", 2: """sequence"""}
            else:
                  lowerCAmelCase  						=       {0: """batch""", 1: """sequence"""}
            return OrderedDict(
                [
                    ("""input_ids""", dynamic_axis),
                    ("""attention_mask""", dynamic_axis),
                ] )














from maths.prime_check import is_prime
def 					snake_case_						(   lowerCAmelCase_				:  int					):
							if not isinstance(lowerCAmelCase_				,				lowerCAmelCase_					):
														__lowercase					:							Dict			=   F"Input value of [number={number}] must be an integer"
														raise TypeError(lowerCAmelCase_					)
							if is_prime(lowerCAmelCase_					) and is_prime(number + 2					):
														return number + 2
							else:
														return -1


if __name__ == "__main__":
	import doctest

	doctest.testmod()