


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


a			:			List[str]												=					logging.get_logger(__name__)

a			:			List[Any]												=					{
    '''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_							(					lowerCAmelCase__					):



					'''simple docstring'''







					__UpperCAmelCase							  =					"vivit"






					def __init__(					self			,							snake_case_=2_2_4			,							snake_case_=3_2			,							snake_case_=[2, 1_6, 1_6]			,							snake_case_=3			,							snake_case_=7_6_8			,							snake_case_=1_2			,							snake_case_=1_2			,							snake_case_=3_0_7_2			,							snake_case_="gelu_fast"			,							snake_case_=0.0			,							snake_case_=0.0			,							snake_case_=0.0_2			,							snake_case_=1e-0_6			,							snake_case_=True			,							**snake_case_			,							) ->      int:







								'''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__(**snake_case_    )






import copy
from dataclasses import dataclass, field
from typing import ClassVar, Dict

from ..features import Audio, Features, Value
from .base import TaskTemplate
@dataclass(frozen=lowerCAmelCase__					)
class 						lowerCamelCase_							(					lowerCAmelCase__					):



					'''simple docstring'''







					__UpperCAmelCase							  =					field(default="automatic-speech-recognition"  ,  metadata={"include_in_asdict_even_if_is_default": True}					)
					__UpperCAmelCase							  =					Features({"audio": Audio()}					)
					__UpperCAmelCase							  =					Features({"transcription": Value("string"					)}					)
					__UpperCAmelCase							  =					"audio"
					__UpperCAmelCase							  =					"transcription"






					def A   (					self			,							snake_case_    ) ->      Any:







								'''simple docstring'''

								if self.audio_column not in features:
											raise ValueError(F'Column {self.audio_column} is not present in features.'    )
								if not isinstance(features[self.audio_column]			,							snake_case_    ):
											raise ValueError(F'Column {self.audio_column} is not an Audio type.'    )
								__lowercase			       = copy.deepcopy(self    )
								__lowercase			       = self.input_schema.copy()
								__lowercase			       = features[self.audio_column]
								__lowercase			       = input_schema
								return task_template






					@property
					def A   (					self    ) ->      Dict[str, str]:







								'''simple docstring'''

								return {self.audio_column: "audio", self.transcription_column: "transcription"}


