






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


__UpperCAmelCase								=						logging.get_logger(__name__)

__UpperCAmelCase								=						{
    '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   lowerCamelCase      (_snake_case			):







		'''simple docstring'''



		_snake_case    :    Tuple       						=  '''vivit'''



		def __init__(							self				,		_UpperCamelCase=2_2_4				,		_UpperCamelCase=3_2				,		_UpperCamelCase=[2, 1_6, 1_6]				,		_UpperCamelCase=3				,		_UpperCamelCase=7_6_8				,		_UpperCamelCase=1_2				,		_UpperCamelCase=1_2				,		_UpperCamelCase=3_0_7_2				,		_UpperCamelCase="gelu_fast"				,		_UpperCamelCase=0.0				,		_UpperCamelCase=0.0				,		_UpperCamelCase=0.02				,		_UpperCamelCase=1E-06				,		_UpperCamelCase=True				,		**_UpperCamelCase				,		)	->							int:
							UpperCAmelCase_    :      Optional[Any]            =						hidden_size
							UpperCAmelCase_    :      Any            =						num_hidden_layers
							UpperCAmelCase_    :      Dict            =						num_attention_heads
							UpperCAmelCase_    :      Any            =						intermediate_size
							UpperCAmelCase_    :      Optional[int]            =						hidden_act
							UpperCAmelCase_    :      List[Any]            =						hidden_dropout_prob
							UpperCAmelCase_    :      List[Any]            =						attention_probs_dropout_prob
							UpperCAmelCase_    :      List[Any]            =						initializer_range
							UpperCAmelCase_    :      List[str]            =						layer_norm_eps

							UpperCAmelCase_    :      List[Any]            =						image_size
							UpperCAmelCase_    :      Dict            =						num_frames
							UpperCAmelCase_    :      int            =						tubelet_size
							UpperCAmelCase_    :      List[str]            =						num_channels
							UpperCAmelCase_    :      str            =						qkv_bias

							super().__init__(**_UpperCamelCase   )







"""simple docstring"""







from argparse import ArgumentParser

from datasets.commands.convert import ConvertCommand
from datasets.commands.dummy_data import DummyDataCommand
from datasets.commands.env import EnvironmentCommand
from datasets.commands.run_beam import RunBeamCommand
from datasets.commands.test import TestCommand
from datasets.utils.logging import set_verbosity_info



def 			lowercase__(       __SCREAMING_SNAKE_CASE							:     Optional[int]      ):
     return {key.lstrip('-'      ): value for key, value in zip(unknown_args[::2]     ,			unknown_args[1::2]      )}



def 			lowercase__(       ):
     lowercase_	:				Any													=	ArgumentParser(
         'HuggingFace Datasets CLI tool'     ,			usage='datasets-cli <command> [<args>]'     ,			allow_abbrev=__SCREAMING_SNAKE_CASE      )
     lowercase_	:				Tuple													=	parser.add_subparsers(help='datasets-cli command helpers'      )
     set_verbosity_info()

     # Register commands
     ConvertCommand.register_subcommand(__SCREAMING_SNAKE_CASE      )
     EnvironmentCommand.register_subcommand(__SCREAMING_SNAKE_CASE      )
     TestCommand.register_subcommand(__SCREAMING_SNAKE_CASE      )
     RunBeamCommand.register_subcommand(__SCREAMING_SNAKE_CASE      )
     DummyDataCommand.register_subcommand(__SCREAMING_SNAKE_CASE      )

     # Parse args
     lowercase_							,  lowercase_	:				Dict													=	parser.parse_known_args()
     if not hasattr(__SCREAMING_SNAKE_CASE     ,			'func'      ):
          parser.print_help()
          exit(1      )
     lowercase_	:				int													=	parse_unknown_args(__SCREAMING_SNAKE_CASE      )

     # Run
     lowercase_	:				List[Any]													=	args.func(__SCREAMING_SNAKE_CASE     ,			**__SCREAMING_SNAKE_CASE      )
     service.run()


if __name__ == "__main__":
       main()

