

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


_SCREAMING_SNAKE_CASE					:  Dict             =      logging.get_logger(__name__)

_SCREAMING_SNAKE_CASE					:  Optional[Any]             =      {
    '''facebook/vit-mae-base''': '''https://huggingface.co/facebook/vit-mae-base/resolve/main/config.json''',
    # See all ViT MAE models at https://huggingface.co/models?filter=vit-mae
}






class 	UpperCAmelCase__		(	__snake_case	):






  """simple docstring"""


  a									= """vit_mae"""






  def __init__(							self						:		Dict     ,       __lowerCamelCase						:		str=768     ,       __lowerCamelCase						:		Any=12     ,       __lowerCamelCase						:		Optional[int]=12     ,       __lowerCamelCase						:		str=3072     ,       __lowerCamelCase						:		Tuple="gelu"     ,       __lowerCamelCase						:		Optional[int]=0.0     ,       __lowerCamelCase						:		Tuple=0.0     ,       __lowerCamelCase						:		Optional[Any]=0.02     ,       __lowerCamelCase						:		Union[str, Any]=1e-12     ,       __lowerCamelCase						:		Dict=224     ,       __lowerCamelCase						:		Optional[Any]=16     ,       __lowerCamelCase						:		List[Any]=3     ,       __lowerCamelCase						:		List[Any]=True     ,       __lowerCamelCase						:		Optional[int]=16     ,       __lowerCamelCase						:		List[Any]=512     ,       __lowerCamelCase						:		str=8     ,       __lowerCamelCase						:		str=2048     ,       __lowerCamelCase						:		int=0.75     ,       __lowerCamelCase						:		Tuple=False     ,       **__lowerCamelCase						:		List[str]     ,       )		->							int:
       super().__init__(**UpperCamelCase__ )

       SCREAMING_SNAKE_CASE__   =   hidden_size
       SCREAMING_SNAKE_CASE__   =   num_hidden_layers
       SCREAMING_SNAKE_CASE__   =   num_attention_heads
       SCREAMING_SNAKE_CASE__   =   intermediate_size
       SCREAMING_SNAKE_CASE__   =   hidden_act
       SCREAMING_SNAKE_CASE__   =   hidden_dropout_prob
       SCREAMING_SNAKE_CASE__   =   attention_probs_dropout_prob
       SCREAMING_SNAKE_CASE__   =   initializer_range
       SCREAMING_SNAKE_CASE__   =   layer_norm_eps
       SCREAMING_SNAKE_CASE__   =   image_size
       SCREAMING_SNAKE_CASE__   =   patch_size
       SCREAMING_SNAKE_CASE__   =   num_channels
       SCREAMING_SNAKE_CASE__   =   qkv_bias
       SCREAMING_SNAKE_CASE__   =   decoder_num_attention_heads
       SCREAMING_SNAKE_CASE__   =   decoder_hidden_size
       SCREAMING_SNAKE_CASE__   =   decoder_num_hidden_layers
       SCREAMING_SNAKE_CASE__   =   decoder_intermediate_size
       SCREAMING_SNAKE_CASE__   =   mask_ratio
       SCREAMING_SNAKE_CASE__   =   norm_pix_loss



import warnings

from .generation import TFGenerationMixin






class 	UpperCAmelCase__		(	A__	):






  """simple docstring"""


  warnings.warn(
      "Importing `TFGenerationMixin` from `src/transformers/generation_tf_utils.py` is deprecated and will "
      "be removed in Transformers v5. Import as `from transformers import TFGenerationMixin` instead."						,					A__						,					)


