

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


lowerCAmelCase_       	=	logging.get_logger(__name__)

lowerCAmelCase_       	=	{
    '''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_   (		__A					):







			'''simple docstring'''

			SCREAMING_SNAKE_CASE     :   int									=				"vivit"







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

									snake_case_   =	image_size
									snake_case_   =	num_frames
									snake_case_   =	tubelet_size
									snake_case_   =	num_channels
									snake_case_   =	qkv_bias

									super().__init__(**_UpperCamelCase							)




"""simple docstring"""






from typing import TYPE_CHECKING

from ..utils import _LazyModule


__snake_case							:   Optional[int]      =    {
    'config': [
        'EXTERNAL_DATA_FORMAT_SIZE_LIMIT',
        'OnnxConfig',
        'OnnxConfigWithPast',
        'OnnxSeq2SeqConfigWithPast',
        'PatchingSpec',
    ],
    'convert': ['export', 'validate_model_outputs'],
    'features': ['FeaturesManager'],
    'utils': ['ParameterFormat', 'compute_serialized_parameters_size'],
}


if TYPE_CHECKING:
	from .config import (
	    EXTERNAL_DATA_FORMAT_SIZE_LIMIT,
	    OnnxConfig,
	    OnnxConfigWithPast,
	    OnnxSeqaSeqConfigWithPast,
	    PatchingSpec,
	)
	from .convert import export, validate_model_outputs
	from .features import FeaturesManager
	from .utils import ParameterFormat, compute_serialized_parameters_size

else:
	import sys

	__snake_case							:   Any      =    _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)