






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


A_						:						Dict						      =					logging.get_logger(__name__)

A_						:						int						      =					{
    '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 							_lowerCAmelCase(			UpperCAmelCase_   ):





       """simple docstring"""







       a							:						Tuple										='''vit_mae'''





       def __init__(   self    ,     _lowerCamelCase=7_6_8    ,     _lowerCamelCase=1_2    ,     _lowerCamelCase=1_2    ,     _lowerCamelCase=3_0_7_2    ,     _lowerCamelCase="gelu"    ,     _lowerCamelCase=0.0    ,     _lowerCamelCase=0.0    ,     _lowerCamelCase=0.0_2    ,     _lowerCamelCase=1e-12    ,     _lowerCamelCase=2_2_4    ,     _lowerCamelCase=1_6    ,     _lowerCamelCase=3    ,     _lowerCamelCase=True    ,     _lowerCamelCase=1_6    ,     _lowerCamelCase=5_1_2    ,     _lowerCamelCase=8    ,     _lowerCamelCase=2_0_4_8    ,     _lowerCamelCase=0.7_5    ,     _lowerCamelCase=False    ,     **_lowerCamelCase    ,     ):
         super().__init__(**_lowerCamelCase				)

         UpperCamelCase_:   Any        =							hidden_size
         UpperCamelCase_:   Any        =							num_hidden_layers
         UpperCamelCase_:   Tuple        =							num_attention_heads
         UpperCamelCase_:   str        =							intermediate_size
         UpperCamelCase_:   Dict        =							hidden_act
         UpperCamelCase_:   Any        =							hidden_dropout_prob
         UpperCamelCase_:   Optional[int]        =							attention_probs_dropout_prob
         UpperCamelCase_:   Any        =							initializer_range
         UpperCamelCase_:   List[str]        =							layer_norm_eps
         UpperCamelCase_:   Union[str, Any]        =							image_size
         UpperCamelCase_:   Any        =							patch_size
         UpperCamelCase_:   Any        =							num_channels
         UpperCamelCase_:   str        =							qkv_bias
         UpperCamelCase_:   str        =							decoder_num_attention_heads
         UpperCamelCase_:   Dict        =							decoder_hidden_size
         UpperCamelCase_:   Dict        =							decoder_num_hidden_layers
         UpperCamelCase_:   Optional[Any]        =							decoder_intermediate_size
         UpperCamelCase_:   List[str]        =							mask_ratio
         UpperCamelCase_:   Tuple        =							norm_pix_loss


def 	__a     (      A__			:  float			,  A__			:  float       ):
    if mass < 0:
        raise ValueError("The mass of a body cannot be negative"       )
    return 0.5 * mass * abs(A__       ) * abs(A__       )


if __name__ == "__main__":
    import doctest

    doctest.testmod(verbose=True)