




'''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


UpperCAmelCase_					=   logging.get_logger(__name__)

UpperCAmelCase_					=   {
    "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  __lowercase    (	__magic_name__			):
   _a    						=   """vit"""







   def __init__(						self   ,				UpperCamelCase=768   ,				UpperCamelCase=12   ,				UpperCamelCase=12   ,				UpperCamelCase=3072   ,				UpperCamelCase="gelu"   ,				UpperCamelCase=0.0   ,				UpperCamelCase=0.0   ,				UpperCamelCase=0.02   ,				UpperCamelCase=1e-12   ,				UpperCamelCase=224   ,				UpperCamelCase=16   ,				UpperCamelCase=3   ,				UpperCamelCase=True   ,				UpperCamelCase=16   ,				**UpperCamelCase   ,				)					->     Tuple:
     super().__init__(**UpperCamelCase   )

     __a								=       hidden_size
     __a								=       num_hidden_layers
     __a								=       num_attention_heads
     __a								=       intermediate_size
     __a								=       hidden_act
     __a								=       hidden_dropout_prob
     __a								=       attention_probs_dropout_prob
     __a								=       initializer_range
     __a								=       layer_norm_eps
     __a								=       image_size
     __a								=       patch_size
     __a								=       num_channels
     __a								=       qkv_bias
     __a								=       encoder_stride





class  __lowercase    (	__magic_name__			):
   _a    						=   version.parse("""1.11"""			)







   @property
   def 		UpperCamelCase__      (						self   )					->     Mapping[str, Mapping[int, str]]:
     return OrderedDict(
         [
             ('pixel_values', {0: 'batch', 1: 'num_channels', 2: 'height', 3: 'width'}),
         ]   )







   @property
   def 		UpperCamelCase__      (						self   )					->     float:
     return 1e-4









'''simple docstring'''


from typing import TYPE_CHECKING

from ....utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


UpperCAmelCase_					=   {
    "configuration_trajectory_transformer": [
        "TRAJECTORY_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP",
        "TrajectoryTransformerConfig",
    ],
}

try:
  if not is_torch_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  UpperCAmelCase_					=   [
      "TRAJECTORY_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
      "TrajectoryTransformerModel",
      "TrajectoryTransformerPreTrainedModel",
      "load_tf_weights_in_trajectory_transformer",
  ]


if TYPE_CHECKING:
  from .configuration_trajectory_transformer import (
      TRAJECTORY_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP,
      TrajectoryTransformerConfig,
  )

  try:
    if not is_torch_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_trajectory_transformer import (
        TRAJECTORY_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
        TrajectoryTransformerModel,
        TrajectoryTransformerPreTrainedModel,
        load_tf_weights_in_trajectory_transformer,
    )


else:
  import sys

  UpperCAmelCase_					=   _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)



