


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


snake_case							: int      =   logging.get_logger(__name__)

snake_case							: Dict      =   {
    '''google/vivit-b-16x2-kinetics400''': (
        '''https://huggingface.co/google/vivit-b-16x2-kinetics400/resolve/main/config.json'''
    ),
    # See all Vivit models at https://huggingface.co/models?filter=vivit
}







class  snake_case_     (lowerCamelCase_   ):
					UpperCAmelCase__      :					Dict					   =    '''vivit'''
					def __init__(   self     :Union[str, Any]    ,__snake_case     :Dict=2_24    ,__snake_case     :Any=32    ,__snake_case     :Any=[2, 16, 16]    ,__snake_case     :Optional[int]=3    ,__snake_case     :Dict=7_68    ,__snake_case     :Any=12    ,__snake_case     :int=12    ,__snake_case     :Tuple=30_72    ,__snake_case     :Optional[Any]="gelu_fast"    ,__snake_case     :Dict=0.0    ,__snake_case     :Union[str, Any]=0.0    ,__snake_case     :str=0.02    ,__snake_case     :Any=1E-06    ,__snake_case     :Any=True    ,**__snake_case     :Optional[Any]    ,)      ->					List[Any]:
												a__    =  hidden_size
												a__    =  num_hidden_layers
												a__    =  num_attention_heads
												a__    =  intermediate_size
												a__    =  hidden_act
												a__    =  hidden_dropout_prob
												a__    =  attention_probs_dropout_prob
												a__    =  initializer_range
												a__    =  layer_norm_eps

												a__    =  image_size
												a__    =  num_frames
												a__    =  tubelet_size
												a__    =  num_channels
												a__    =  qkv_bias

												super().__init__(**__snake_case   )








def 							__lowercase     (				__lowerCAmelCase							:       int     ):

							if length <= 0 or not isinstance(__lowerCAmelCase      ,					__lowerCAmelCase     ):
														raise ValueError('Length must be a positive integer.'     )
							return [n * (2 * n - 1) for n in range(__lowerCAmelCase     )]


if __name__ == "__main__":
						print(hexagonal_numbers(length=5))
						print(hexagonal_numbers(length=10))




