
'''simple docstring'''







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


_lowercase			:  Any   							=	logging.get_logger(__name__)

_lowercase			:  Optional[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__       (  lowerCamelCase_					):
					lowerCAmelCase_					       =					'''vit_mae'''






					def __init__(       self    ,							__SCREAMING_SNAKE_CASE=7_68    ,							__SCREAMING_SNAKE_CASE=12    ,							__SCREAMING_SNAKE_CASE=12    ,							__SCREAMING_SNAKE_CASE=30_72    ,							__SCREAMING_SNAKE_CASE="gelu"    ,							__SCREAMING_SNAKE_CASE=0.0    ,							__SCREAMING_SNAKE_CASE=0.0    ,							__SCREAMING_SNAKE_CASE=0.02    ,							__SCREAMING_SNAKE_CASE=1E-1_2    ,							__SCREAMING_SNAKE_CASE=2_24    ,							__SCREAMING_SNAKE_CASE=16    ,							__SCREAMING_SNAKE_CASE=3    ,							__SCREAMING_SNAKE_CASE=True    ,							__SCREAMING_SNAKE_CASE=16    ,							__SCREAMING_SNAKE_CASE=5_12    ,							__SCREAMING_SNAKE_CASE=8    ,							__SCREAMING_SNAKE_CASE=20_48    ,							__SCREAMING_SNAKE_CASE=0.75    ,							__SCREAMING_SNAKE_CASE=False    ,							**__SCREAMING_SNAKE_CASE    ,							):


											"""simple docstring"""





											super().__init__(**__SCREAMING_SNAKE_CASE		)

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





'''simple docstring'''







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_lowercase			:  Tuple   							=	{"configuration_vit_msn": ["VIT_MSN_PRETRAINED_CONFIG_ARCHIVE_MAP", "ViTMSNConfig"]}

try:
							if not is_torch_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							pass
else:
							_lowercase			:  str   							=	[
							    "VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST",
							    "ViTMSNModel",
							    "ViTMSNForImageClassification",
							    "ViTMSNPreTrainedModel",
							]

if TYPE_CHECKING:
							from .configuration_vit_msn import VIT_MSN_PRETRAINED_CONFIG_ARCHIVE_MAP, ViTMSNConfig

							try:
														if not is_torch_available():
																					raise OptionalDependencyNotAvailable()
							except OptionalDependencyNotAvailable:
														pass
							else:
														from .modeling_vit_msn import (
														    VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST,
														    ViTMSNForImageClassification,
														    ViTMSNModel,
														    ViTMSNPreTrainedModel,
														)

else:
							import sys

							_lowercase			:  int   							=	_LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)



