


'''simple docstring'''




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


UpperCAmelCase_      :		Optional[int]          =		logging.get_logger(__name__)

UpperCAmelCase_      :		List[str]          =		{
    "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 					a			(				snake_case__  ):
 '''simple docstring'''




 __lowerCAmelCase						: str							  =		"""vivit"""




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

    _a					:  Any						=  image_size
    _a					:  Optional[Any]						=  num_frames
    _a					:  Dict						=  tubelet_size
    _a					:  Union[str, Any]						=  num_channels
    _a					:  Optional[int]						=  qkv_bias

    super().__init__(**lowerCamelCase_      )






'''simple docstring'''




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


UpperCAmelCase_      :		Optional[int]          =		logging.get_logger(__name__)

UpperCAmelCase_      :		List[str]          =		{
    "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 					a			(				snake_case__  ):
 '''simple docstring'''




 __lowerCAmelCase						: str							  =		"""vivit"""




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

    _a					:  Any						=  image_size
    _a					:  Optional[Any]						=  num_frames
    _a					:  Dict						=  tubelet_size
    _a					:  Union[str, Any]						=  num_channels
    _a					:  Optional[int]						=  qkv_bias

    super().__init__(**lowerCamelCase_      )


