






"""simple docstring"""





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


UpperCamelCase					=     logging.get_logger(__name__)

UpperCamelCase					=     {
    """ZinengTang/tvlt-base""": """https://huggingface.co/ZinengTang/tvlt-base/blob/main/config.json""",
}







class 						_a  (		lowercase_       ):





   '''simple docstring'''






   UpperCamelCase__     	=							"""tvlt"""





   def __init__(       self					, UpperCAmelCase_=224					, UpperCAmelCase_=2_048					, UpperCAmelCase_=128					, UpperCAmelCase_=[16, 16]					, UpperCAmelCase_=[16, 16]					, UpperCAmelCase_=3					, UpperCAmelCase_=1					, UpperCAmelCase_=8					, UpperCAmelCase_=768					, UpperCAmelCase_=12					, UpperCAmelCase_=12					, UpperCAmelCase_=3_072					, UpperCAmelCase_="gelu"					, UpperCAmelCase_=0.0					, UpperCAmelCase_=0.0					, UpperCAmelCase_=0.02					, UpperCAmelCase_=1E-6					, UpperCAmelCase_=True					, UpperCAmelCase_=False					, UpperCAmelCase_=16					, UpperCAmelCase_=512					, UpperCAmelCase_=8					, UpperCAmelCase_=2_048					, UpperCAmelCase_=0.75					, UpperCAmelCase_=0.15					, UpperCAmelCase_="frame-level"					, UpperCAmelCase_=True					, UpperCAmelCase_=True					, UpperCAmelCase_="classification"					, **UpperCAmelCase_					, )			->   Tuple:




          '''simple docstring'''




          super().__init__(**UpperCAmelCase_)

          if audio_mask_type not in ("frame-level", "patch_level"):
                 raise ValueError(
                     "audio_mask_type must be one of two acceptable strategies - {'frame_level', 'patch-level') "
                     F"""got {audio_mask_type}""")

          lowercase__:							Optional[int]         =     image_size
          lowercase__:							Dict         =     spectrogram_length
          lowercase__:							List[str]         =     frequency_length
          lowercase__:							int         =     image_patch_size
          lowercase__:							Optional[int]         =     audio_patch_size
          lowercase__:							Tuple         =     num_image_channels
          lowercase__:							List[str]         =     num_audio_channels
          lowercase__:							Tuple         =     num_frames

          lowercase__:							List[str]         =     hidden_size
          lowercase__:							str         =     num_hidden_layers
          lowercase__:							List[str]         =     num_attention_heads
          lowercase__:							List[Any]         =     intermediate_size
          lowercase__:							Union[str, Any]         =     hidden_act
          lowercase__:							Optional[int]         =     hidden_dropout_prob
          lowercase__:							Optional[Any]         =     attention_probs_dropout_prob
          lowercase__:							Union[str, Any]         =     initializer_range
          lowercase__:							Any         =     layer_norm_eps
          lowercase__:							Any         =     qkv_bias
          lowercase__:							List[str]         =     use_mean_pooling

          lowercase__:							Tuple         =     decoder_num_attention_heads
          lowercase__:							Any         =     decoder_hidden_size
          lowercase__:							Any         =     decoder_num_hidden_layers
          lowercase__:							Union[str, Any]         =     decoder_intermediate_size
          lowercase__:							Optional[Any]         =     pixel_mask_ratio
          lowercase__:							str         =     audio_mask_ratio
          lowercase__:							Dict         =     audio_mask_type

          lowercase__:							Union[str, Any]         =     task_matching
          lowercase__:							Union[str, Any]         =     task_mae
          lowercase__:							Union[str, Any]         =     loss_type







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


UpperCamelCase    				=		{'''configuration_dinat''': ['''DINAT_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''DinatConfig''']}


try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase    				=		[
          '''DINAT_PRETRAINED_MODEL_ARCHIVE_LIST''',
          '''DinatForImageClassification''',
          '''DinatModel''',
          '''DinatPreTrainedModel''',
          '''DinatBackbone''',
      ]

if TYPE_CHECKING:
      from .configuration_dinat import DINAT_PRETRAINED_CONFIG_ARCHIVE_MAP, DinatConfig

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_dinat import (
                DINAT_PRETRAINED_MODEL_ARCHIVE_LIST,
                DinatBackbone,
                DinatForImageClassification,
                DinatModel,
                DinatPreTrainedModel,
            )

else:
      import sys

      UpperCamelCase    				=		_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)



