

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


UpperCAmelCase__			:  str					     =     logging.get_logger(__name__)

UpperCAmelCase__			:  Dict					     =     {
    """uw-madison/yoso-4096""": """https://huggingface.co/uw-madison/yoso-4096/resolve/main/config.json""",
    # See all YOSO models at https://huggingface.co/models?filter=yoso
}







class 							a__							( UpperCAmelCase							):







  """simple docstring"""




  UpperCAmelCase__   :			Union[str, Any]      							="""yoso"""


  def __init__(      self     :			List[str]					,					UpperCAmelCase__     :			Any=5_0_2_6_5					,					UpperCAmelCase__     :			Optional[Any]=7_6_8					,					UpperCAmelCase__     :			int=1_2					,					UpperCAmelCase__     :			Dict=1_2					,					UpperCAmelCase__     :			Tuple=3_0_7_2					,					UpperCAmelCase__     :			List[str]="gelu"					,					UpperCAmelCase__     :			int=0.1					,					UpperCAmelCase__     :			int=0.1					,					UpperCAmelCase__     :			Any=4_0_9_6					,					UpperCAmelCase__     :			Union[str, Any]=1					,					UpperCAmelCase__     :			Optional[int]=0.02					,					UpperCAmelCase__     :			List[Any]=1e-12					,					UpperCAmelCase__     :			Tuple="absolute"					,					UpperCAmelCase__     :			Optional[Any]=True					,					UpperCAmelCase__     :			Union[str, Any]=9					,					UpperCAmelCase__     :			List[Any]=6_4					,					UpperCAmelCase__     :			Optional[Any]=None					,					UpperCAmelCase__     :			str=True					,					UpperCAmelCase__     :			Optional[int]=True					,					UpperCAmelCase__     :			Dict=1					,					UpperCAmelCase__     :			Union[str, Any]=0					,					UpperCAmelCase__     :			List[Any]=2					,					**UpperCAmelCase__     :			Tuple					,					)       ->str:
        """simple docstring"""







        super().__init__(pad_token_id=UpperCAmelCase__					,					bos_token_id=UpperCAmelCase__					,					eos_token_id=UpperCAmelCase__					,					**UpperCAmelCase__		)

        SCREAMING_SNAKE_CASE			:   List[Any]	  = vocab_size
        SCREAMING_SNAKE_CASE			:   List[Any]	  = max_position_embeddings
        SCREAMING_SNAKE_CASE			:   List[str]	  = hidden_size
        SCREAMING_SNAKE_CASE			:   Dict	  = num_hidden_layers
        SCREAMING_SNAKE_CASE			:   str	  = num_attention_heads
        SCREAMING_SNAKE_CASE			:   int	  = intermediate_size
        SCREAMING_SNAKE_CASE			:   Any	  = hidden_act
        SCREAMING_SNAKE_CASE			:   int	  = hidden_dropout_prob
        SCREAMING_SNAKE_CASE			:   Union[str, Any]	  = attention_probs_dropout_prob
        SCREAMING_SNAKE_CASE			:   Tuple	  = initializer_range
        SCREAMING_SNAKE_CASE			:   Dict	  = type_vocab_size
        SCREAMING_SNAKE_CASE			:   Tuple	  = layer_norm_eps
        SCREAMING_SNAKE_CASE			:   Dict	  = position_embedding_type
        SCREAMING_SNAKE_CASE			:   Any	  = use_expectation
        SCREAMING_SNAKE_CASE			:   str	  = hash_code_len
        SCREAMING_SNAKE_CASE			:   str	  = num_hash
        SCREAMING_SNAKE_CASE			:   str	  = conv_window
        SCREAMING_SNAKE_CASE			:   int	  = use_fast_hash
        SCREAMING_SNAKE_CASE			:   str	  = lsh_backward







import itertools






def        __lowercase   (  _A = 1000000      )  ->    int:
      SCREAMING_SNAKE_CASE			:   Tuple	  = [1]

      for i in itertools.count(len(_A      )      ):
            SCREAMING_SNAKE_CASE			:   Optional[int]	  = 0
            for j in itertools.count(1      ):
                  SCREAMING_SNAKE_CASE			:   str	  = -1 if j % 2 == 0 else +1
                  SCREAMING_SNAKE_CASE			:   Any	  = (j * j * 3 - j) // 2
                  if index > i:
                        break
                  item += partitions[i - index] * sign
                  item %= number
                  index += j
                  if index > i:
                        break
                  item += partitions[i - index] * sign
                  item %= number

            if item == 0:
                  return i
            partitions.append(_A      )

      return 0


if __name__ == "__main__":
  import doctest

  doctest.testmod()

  print(F"""{solution() = }""")




