'''simple docstring'''





from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_tf_available,
    is_torch_available,
    is_vision_available,
)


UpperCamelCase       :     str			=    {
    """configuration_segformer""": ["""SEGFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP""", """SegformerConfig""", """SegformerOnnxConfig"""]
}

try:
      if not is_vision_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase       :     Any			=    ["""SegformerFeatureExtractor"""]
      UpperCamelCase       :     Optional[Any]			=    ["""SegformerImageProcessor"""]

try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase       :     str			=    [
          """SEGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST""",
          """SegformerDecodeHead""",
          """SegformerForImageClassification""",
          """SegformerForSemanticSegmentation""",
          """SegformerLayer""",
          """SegformerModel""",
          """SegformerPreTrainedModel""",
      ]

try:
      if not is_tf_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase       :     Optional[int]			=    [
          """TF_SEGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST""",
          """TFSegformerDecodeHead""",
          """TFSegformerForImageClassification""",
          """TFSegformerForSemanticSegmentation""",
          """TFSegformerModel""",
          """TFSegformerPreTrainedModel""",
      ]


if TYPE_CHECKING:
      from .configuration_segformer import SEGFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, SegformerConfig, SegformerOnnxConfig

      try:
            if not is_vision_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .feature_extraction_segformer import SegformerFeatureExtractor
            from .image_processing_segformer import SegformerImageProcessor

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_segformer import (
                SEGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
                SegformerDecodeHead,
                SegformerForImageClassification,
                SegformerForSemanticSegmentation,
                SegformerLayer,
                SegformerModel,
                SegformerPreTrainedModel,
            )
      try:
            if not is_tf_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_tf_segformer import (
                TF_SEGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
                TFSegformerDecodeHead,
                TFSegformerForImageClassification,
                TFSegformerForSemanticSegmentation,
                TFSegformerModel,
                TFSegformerPreTrainedModel,
            )

else:
      import sys

      UpperCamelCase       :     List[Any]			=    _LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)


'''simple docstring'''





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


UpperCamelCase       :     Optional[Any]			=    logging.get_logger(__name__)

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



class   UpperCamelCase					(       a_       ):







    """simple docstring"""





    A     :    int				=			"tvlt"





    def __init__(					self     :						List[str]							,							UpperCAmelCase_     :						Any=2_2_4							,							UpperCAmelCase_     :						Optional[Any]=2_0_4_8							,							UpperCAmelCase_     :						Dict=1_2_8							,							UpperCAmelCase_     :						str=[1_6, 1_6]							,							UpperCAmelCase_     :						int=[1_6, 1_6]							,							UpperCAmelCase_     :						int=3							,							UpperCAmelCase_     :						Optional[int]=1							,							UpperCAmelCase_     :						List[str]=8							,							UpperCAmelCase_     :						List[Any]=7_6_8							,							UpperCAmelCase_     :						int=1_2							,							UpperCAmelCase_     :						List[Any]=1_2							,							UpperCAmelCase_     :						Optional[Any]=3_0_7_2							,							UpperCAmelCase_     :						List[str]="gelu"							,							UpperCAmelCase_     :						Tuple=0.0							,							UpperCAmelCase_     :						Optional[Any]=0.0							,							UpperCAmelCase_     :						Tuple=0.02							,							UpperCAmelCase_     :						Any=1e-6							,							UpperCAmelCase_     :						Dict=True							,							UpperCAmelCase_     :						Union[str, Any]=False							,							UpperCAmelCase_     :						Tuple=1_6							,							UpperCAmelCase_     :						str=5_1_2							,							UpperCAmelCase_     :						List[str]=8							,							UpperCAmelCase_     :						Union[str, Any]=2_0_4_8							,							UpperCAmelCase_     :						Tuple=0.75							,							UpperCAmelCase_     :						Optional[Any]=0.15							,							UpperCAmelCase_     :						Union[str, Any]="frame-level"							,							UpperCAmelCase_     :						List[str]=True							,							UpperCAmelCase_     :						Any=True							,							UpperCAmelCase_     :						Optional[int]="classification"							,							**UpperCAmelCase_     :						Any							,							):


         """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}""")

         a					:							str							= image_size
         a					:							List[Any]							= spectrogram_length
         a					:							Union[str, Any]							= frequency_length
         a					:							str							= image_patch_size
         a					:							Dict							= audio_patch_size
         a					:							Dict							= num_image_channels
         a					:							Tuple							= num_audio_channels
         a					:							Optional[int]							= num_frames

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

         a					:							List[Any]							= decoder_num_attention_heads
         a					:							Any							= decoder_hidden_size
         a					:							int							= decoder_num_hidden_layers
         a					:							Tuple							= decoder_intermediate_size
         a					:							Tuple							= pixel_mask_ratio
         a					:							List[Any]							= audio_mask_ratio
         a					:							Optional[Any]							= audio_mask_type

         a					:							Any							= task_matching
         a					:							Optional[Any]							= task_mae
         a					:							List[str]							= loss_type

