
'''simple docstring'''







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_lowercase			:  Any   							=	{
    "configuration_jukebox": [
        "JUKEBOX_PRETRAINED_CONFIG_ARCHIVE_MAP",
        "JukeboxConfig",
        "JukeboxPriorConfig",
        "JukeboxVQVAEConfig",
    ],
    "tokenization_jukebox": ["JukeboxTokenizer"],
}

try:
							if not is_torch_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							pass
else:
							_lowercase			:  Any   							=	[
							    "JUKEBOX_PRETRAINED_MODEL_ARCHIVE_LIST",
							    "JukeboxModel",
							    "JukeboxPreTrainedModel",
							    "JukeboxVQVAE",
							    "JukeboxPrior",
							]

if TYPE_CHECKING:
							from .configuration_jukebox import (
							    JUKEBOX_PRETRAINED_CONFIG_ARCHIVE_MAP,
							    JukeboxConfig,
							    JukeboxPriorConfig,
							    JukeboxVQVAEConfig,
							)
							from .tokenization_jukebox import JukeboxTokenizer

							try:
														if not is_torch_available():
																					raise OptionalDependencyNotAvailable()
							except OptionalDependencyNotAvailable:
														pass
							else:
														from .modeling_jukebox import (
														    JUKEBOX_PRETRAINED_MODEL_ARCHIVE_LIST,
														    JukeboxModel,
														    JukeboxPreTrainedModel,
														    JukeboxPrior,
														    JukeboxVQVAE,
														)

else:
							import sys

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





'''simple docstring'''







from collections import OrderedDict
from typing import Mapping

from packaging import version

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


_lowercase			:  List[Any]   							=	logging.get_logger(__name__)

_lowercase			:  Optional[Any]   							=	{
    "google/vit-base-patch16-224": "https://huggingface.co/vit-base-patch16-224/resolve/main/config.json",
    # See all ViT models at https://huggingface.co/models?filter=vit
}






class 			lowerCAmelCase__       (  lowerCamelCase_					):
					lowerCAmelCase_					       =					'''vit'''






					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    ,							):


											"""simple docstring"""





											super().__init__(**__SCREAMING_SNAKE_CASE		)

											lowercase_     :					Union[str, Any]													=	hidden_size
											lowercase_     :					Dict													=	num_hidden_layers
											lowercase_     :					List[Any]													=	num_attention_heads
											lowercase_     :					Any													=	intermediate_size
											lowercase_     :					Union[str, Any]													=	hidden_act
											lowercase_     :					Dict													=	hidden_dropout_prob
											lowercase_     :					List[str]													=	attention_probs_dropout_prob
											lowercase_     :					Any													=	initializer_range
											lowercase_     :					Tuple													=	layer_norm_eps
											lowercase_     :					Union[str, Any]													=	image_size
											lowercase_     :					Tuple													=	patch_size
											lowercase_     :					Tuple													=	num_channels
											lowercase_     :					Union[str, Any]													=	qkv_bias
											lowercase_     :					List[Any]													=	encoder_stride







class 			lowerCAmelCase__       (  lowerCamelCase_					):
					lowerCAmelCase_					       =					version.parse('''1.11'''					)






					@property
					def      _snake_case   (       self		):


											"""simple docstring"""





											return OrderedDict(
											    [
											        ('''pixel_values''', {0: '''batch''', 1: '''num_channels''', 2: '''height''', 3: '''width'''}),
											    ]		)






					@property
					def      _snake_case   (       self		):


											"""simple docstring"""





											return 1E-4



