





'''simple docstring'''



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


_SCREAMING_SNAKE_CASE					=    logging.get_logger(__name__)

_SCREAMING_SNAKE_CASE					=    {
    '''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 							__lowercase       (      lowerCAmelCase__						):







	'''simple docstring'''

	a    :				int											=       "vivit"


	def __init__(self     ,_lowerCamelCase=224     ,_lowerCamelCase=32     ,_lowerCamelCase=[2, 16, 16]     ,_lowerCamelCase=3     ,_lowerCamelCase=768     ,_lowerCamelCase=12     ,_lowerCamelCase=12     ,_lowerCamelCase=3072     ,_lowerCamelCase="gelu_fast"     ,_lowerCamelCase=0.0     ,_lowerCamelCase=0.0     ,_lowerCamelCase=0.0_2     ,_lowerCamelCase=1E-0_6     ,_lowerCamelCase=True     ,**_lowerCamelCase     ,)							->							List[str]:







				'''simple docstring'''





				__lowercase										=							hidden_size
				__lowercase										=							num_hidden_layers
				__lowercase										=							num_attention_heads
				__lowercase										=							intermediate_size
				__lowercase										=							hidden_act
				__lowercase										=							hidden_dropout_prob
				__lowercase										=							attention_probs_dropout_prob
				__lowercase										=							initializer_range
				__lowercase										=							layer_norm_eps

				__lowercase										=							image_size
				__lowercase										=							num_frames
				__lowercase										=							tubelet_size
				__lowercase										=							num_channels
				__lowercase										=							qkv_bias

				super().__init__(**_lowerCamelCase )












'''simple docstring'''



from argparse import ArgumentParser

from .env import EnvironmentCommand




def    _lowerCAmelCase						(				):
			__lowercase										=							ArgumentParser('''Diffusers CLI tool'''		,       usage='''diffusers-cli <command> [<args>]'''				)
			__lowercase										=							parser.add_subparsers(help='''diffusers-cli command helpers'''				)

			# Register commands
			EnvironmentCommand.register_subcommand(lowerCamelCase_				)

			# Let's go
			__lowercase										=							parser.parse_args()

			if not hasattr(lowerCamelCase_		,       '''func'''				):
						parser.print_help()
						exit(1				)

			# Run
			__lowercase										=							args.func(lowerCamelCase_				)
			service.run()


if __name__ == "__main__":
	main()





