



'''simple docstring'''




import copy
import os
from typing import Union

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


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

lowerCamelCase							:Dict       = {
    '''microsoft/git-base''': '''https://huggingface.co/microsoft/git-base/resolve/main/config.json''',
}







class      _lowerCAmelCase							(      __UpperCAmelCase			):
	__SCREAMING_SNAKE_CASE       :		str		      = 'git_vision_model'


	def __init__(self	,				lowercase=768	,				lowercase=3072	,				lowercase=12	,				lowercase=12	,				lowercase=3	,				lowercase=224	,				lowercase=16	,				lowercase="quick_gelu"	,				lowercase=1E-5	,				lowercase=0.0	,				lowercase=0.02	,				**lowercase	,				):
				super().__init__(**lowercase							)

				A_     :						int     =					hidden_size
				A_     :						str     =					intermediate_size
				A_     :						int     =					num_hidden_layers
				A_     :						Optional[Any]     =					num_attention_heads
				A_     :						Any     =					num_channels
				A_     :						List[Any]     =					patch_size
				A_     :						List[Any]     =					image_size
				A_     :						List[str]     =					initializer_range
				A_     :						List[str]     =					attention_dropout
				A_     :						int     =					layer_norm_eps
				A_     :						str     =					hidden_act


	@classmethod
	def    _a     (cls	,				lowercase	,				**lowercase							):
				cls._set_token_in_kwargs(lowercase							)

				A_     :						Dict     =					cls.get_config_dict(lowercase	,				**lowercase							)

				# get the vision config dict if we are loading from GITConfig
				if config_dict.get("""model_type"""							) == "git":
							A_     :						Tuple     =					config_dict["""vision_config"""]

				if "model_type" in config_dict and hasattr(cls	,				"""model_type"""							) and config_dict["model_type"] != cls.model_type:
							logger.warning(
							    F'You are using a model of type {config_dict["model_type"]} to instantiate a model of type '
							    F'{cls.model_type}. This is not supported for all configurations of models and can yield errors.'							)

				return cls.from_dict(lowercase	,				**lowercase							)







class      _lowerCAmelCase							(      __UpperCAmelCase			):
	__SCREAMING_SNAKE_CASE       :		int		      = 'git'


	def __init__(self	,				lowercase=None	,				lowercase=30522	,				lowercase=768	,				lowercase=6	,				lowercase=12	,				lowercase=3072	,				lowercase="gelu"	,				lowercase=0.1	,				lowercase=0.1	,				lowercase=1024	,				lowercase=0.02	,				lowercase=1E-12	,				lowercase=0	,				lowercase="absolute"	,				lowercase=True	,				lowercase=False	,				lowercase=101	,				lowercase=102	,				lowercase=None	,				**lowercase	,				):
				super().__init__(bos_token_id=lowercase	,				eos_token_id=lowercase	,				pad_token_id=lowercase	,				**lowercase							)

				if vision_config is None:
							A_     :						Optional[Any]     =					{}
							logger.info("""vision_config is None. initializing the GitVisionConfig with default values."""							)

				A_     :						List[str]     =					GitVisionConfig(**lowercase							)
				A_     :						str     =					vocab_size
				A_     :						Dict     =					hidden_size
				A_     :						Union[str, Any]     =					num_hidden_layers
				A_     :						Any     =					num_attention_heads
				A_     :						Dict     =					hidden_act
				A_     :						List[str]     =					intermediate_size
				A_     :						Tuple     =					hidden_dropout_prob
				A_     :						Dict     =					attention_probs_dropout_prob
				A_     :						Tuple     =					max_position_embeddings
				A_     :						List[Any]     =					initializer_range
				A_     :						Optional[Any]     =					layer_norm_eps
				A_     :						List[str]     =					position_embedding_type
				A_     :						int     =					use_cache
				A_     :						int     =					tie_word_embeddings
				A_     :						List[str]     =					num_image_with_embedding

				A_     :						Tuple     =					bos_token_id
				A_     :						Tuple     =					eos_token_id


	def    _a     (self							):
				A_     :						Tuple     =					copy.deepcopy(self.__dict__							)
				A_     :						Tuple     =					self.vision_config.to_dict()
				A_     :						Dict     =					self.__class__.model_type
				return output




'''simple docstring'''




import inspect
import unittest

from transformers import ViTConfig
from transformers.testing_utils import (
    require_accelerate,
    require_torch,
    require_torch_gpu,
    require_vision,
    slow,
    torch_device,
)
from transformers.utils import cached_property, is_torch_available, is_vision_available

from ...test_configuration_common import ConfigTester
from ...test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor
from ...test_pipeline_mixin import PipelineTesterMixin


if is_torch_available():
						import torch
						from torch import nn

						from transformers import ViTForImageClassification, ViTForMaskedImageModeling, ViTModel
						from transformers.models.vit.modeling_vit import VIT_PRETRAINED_MODEL_ARCHIVE_LIST


if is_vision_available():
						from PIL import Image

						from transformers import ViTImageProcessor







class      _lowerCAmelCase							:


	def __init__(self	,				lowercase	,				lowercase=13	,				lowercase=30	,				lowercase=2	,				lowercase=3	,				lowercase=True	,				lowercase=True	,				lowercase=32	,				lowercase=5	,				lowercase=4	,				lowercase=37	,				lowercase="gelu"	,				lowercase=0.1	,				lowercase=0.1	,				lowercase=10	,				lowercase=0.02	,				lowercase=None	,				lowercase=2	,				):
				A_     :						List[str]     =					parent
				A_     :						str     =					batch_size
				A_     :						Optional[Any]     =					image_size
				A_     :						List[str]     =					patch_size
				A_     :						List[str]     =					num_channels
				A_     :						List[str]     =					is_training
				A_     :						str     =					use_labels
				A_     :						List[str]     =					hidden_size
				A_     :						List[Any]     =					num_hidden_layers
				A_     :						Any     =					num_attention_heads
				A_     :						Any     =					intermediate_size
				A_     :						Optional[Any]     =					hidden_act
				A_     :						Optional[int]     =					hidden_dropout_prob
				A_     :						str     =					attention_probs_dropout_prob
				A_     :						Optional[int]     =					type_sequence_label_size
				A_     :						Any     =					initializer_range
				A_     :						int     =					scope
				A_     :						str     =					encoder_stride

				# in ViT, the seq length equals the number of patches + 1 (we add 1 for the [CLS] token)
				A_     :						Dict     =					(image_size // patch_size) ** 2
				A_     :						List[str]     =					num_patches + 1


	def    _a     (self							):
				A_     :						str     =					floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size]							)

				A_     :						Optional[Any]     =					None
				if self.use_labels:
							A_     :						str     =					ids_tensor([self.batch_size]	,				self.type_sequence_label_size							)

				A_     :						Union[str, Any]     =					self.get_config()

				return config, pixel_values, labels


	def    _a     (self							):
				return ViTConfig(
				    image_size=self.image_size	,				patch_size=self.patch_size	,				num_channels=self.num_channels	,				hidden_size=self.hidden_size	,				num_hidden_layers=self.num_hidden_layers	,				num_attention_heads=self.num_attention_heads	,				intermediate_size=self.intermediate_size	,				hidden_act=self.hidden_act	,				hidden_dropout_prob=self.hidden_dropout_prob	,				attention_probs_dropout_prob=self.attention_probs_dropout_prob	,				is_decoder=lowercase	,				initializer_range=self.initializer_range	,				encoder_stride=self.encoder_stride	,				)


	def    _a     (self	,				lowercase	,				lowercase	,				lowercase							):
				A_     :						List[str]     =					ViTModel(config=lowercase							)
				model.to(lowercase							)
				model.eval()
				A_     :						List[str]     =					model(lowercase							)
				self.parent.assertEqual(result.last_hidden_state.shape	,				(self.batch_size, self.seq_length, self.hidden_size)							)


	def    _a     (self	,				lowercase	,				lowercase	,				lowercase							):
				A_     :						List[str]     =					ViTForMaskedImageModeling(config=lowercase							)
				model.to(lowercase							)
				model.eval()
				A_     :						Tuple     =					model(lowercase							)
				self.parent.assertEqual(
				    result.reconstruction.shape	,				(self.batch_size, self.num_channels, self.image_size, self.image_size)							)

				# test greyscale images
				A_     :						Union[str, Any]     =					1
				A_     :						Any     =					ViTForMaskedImageModeling(lowercase							)
				model.to(lowercase							)
				model.eval()

				A_     :						str     =					floats_tensor([self.batch_size, 1, self.image_size, self.image_size]							)
				A_     :						Optional[int]     =					model(lowercase							)
				self.parent.assertEqual(result.reconstruction.shape	,				(self.batch_size, 1, self.image_size, self.image_size)							)


	def    _a     (self	,				lowercase	,				lowercase	,				lowercase							):
				A_     :						Dict     =					self.type_sequence_label_size
				A_     :						str     =					ViTForImageClassification(lowercase							)
				model.to(lowercase							)
				model.eval()
				A_     :						List[str]     =					model(lowercase	,				labels=lowercase							)
				self.parent.assertEqual(result.logits.shape	,				(self.batch_size, self.type_sequence_label_size)							)

				# test greyscale images
				A_     :						Any     =					1
				A_     :						str     =					ViTForImageClassification(lowercase							)
				model.to(lowercase							)
				model.eval()

				A_     :						Optional[int]     =					floats_tensor([self.batch_size, 1, self.image_size, self.image_size]							)
				A_     :						Union[str, Any]     =					model(lowercase							)
				self.parent.assertEqual(result.logits.shape	,				(self.batch_size, self.type_sequence_label_size)							)


	def    _a     (self							):
				A_     :						str     =					self.prepare_config_and_inputs()
				(
				    (
				    A_
				),   (
				    A_
				),   (
				    A_
				),   
				)     :						Optional[int]     =					config_and_inputs
				A_     :						Tuple     =					{"""pixel_values""": pixel_values}
				return config, inputs_dict







@require_torch
class      _lowerCAmelCase							(      __UpperCAmelCase				,      __UpperCAmelCase				,      unittest.TestCase			):
	__SCREAMING_SNAKE_CASE       :		Any		      = (
	    (
	        ViTModel,
	        ViTForImageClassification,
	        ViTForMaskedImageModeling,
	    )
	    if is_torch_available()
	    else ()
	)
	__SCREAMING_SNAKE_CASE       :		Union[str, Any]		      = (
	    {'feature-extraction': ViTModel, 'image-classification': ViTForImageClassification}
	    if is_torch_available()
	    else {}
	)
	__SCREAMING_SNAKE_CASE       :		Union[str, Any]		      = True

	__SCREAMING_SNAKE_CASE       :		Dict		      = False
	__SCREAMING_SNAKE_CASE       :		Optional[int]		      = False
	__SCREAMING_SNAKE_CASE       :		Optional[int]		      = False


	def    _a     (self							):
				A_     :						Any     =					ViTModelTester(self							)
				A_     :						str     =					ConfigTester(self	,				config_class=lowercase	,				has_text_modality=lowercase	,				hidden_size=37							)


	def    _a     (self							):
				self.config_tester.run_common_tests()


	@unittest.skip(reason="""ViT does not use inputs_embeds"""							)
	def    _a     (self							):
				pass


	def    _a     (self							):
				A_,   A_     :						Union[str, Any]     =					self.model_tester.prepare_config_and_inputs_for_common()

				for model_class in self.all_model_classes:
							A_     :						Union[str, Any]     =					model_class(lowercase							)
							self.assertIsInstance(model.get_input_embeddings()	,				(nn.Module)							)
							A_     :						Tuple     =					model.get_output_embeddings()
							self.assertTrue(x is None or isinstance(lowercase	,				nn.Linear							)							)


	def    _a     (self							):
				A_,   A_     :						List[Any]     =					self.model_tester.prepare_config_and_inputs_for_common()

				for model_class in self.all_model_classes:
							A_     :						Optional[Any]     =					model_class(lowercase							)
							A_     :						Any     =					inspect.signature(model.forward							)
							# signature.parameters is an OrderedDict => so arg_names order is deterministic
							A_     :						List[str]     =					[*signature.parameters.keys()]

							A_     :						Dict     =					["""pixel_values"""]
							self.assertListEqual(arg_names[:1]	,				lowercase							)


	def    _a     (self							):
				A_     :						Tuple     =					self.model_tester.prepare_config_and_inputs()
				self.model_tester.create_and_check_model(*lowercase							)


	def    _a     (self							):
				A_     :						List[Any]     =					self.model_tester.prepare_config_and_inputs()
				self.model_tester.create_and_check_for_masked_image_modeling(*lowercase							)


	def    _a     (self							):
				A_     :						Dict     =					self.model_tester.prepare_config_and_inputs()
				self.model_tester.create_and_check_for_image_classification(*lowercase							)


	@slow
	def    _a     (self							):
				for model_name in VIT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
							A_     :						Dict     =					ViTModel.from_pretrained(lowercase							)
							self.assertIsNotNone(lowercase							)






def      a      (			):
			'''simple docstring'''





			A_     :						List[Any]     =					Image.open("""./tests/fixtures/tests_samples/COCO/000000039769.png"""   )
			return image







@require_torch
@require_vision
class      _lowerCAmelCase							(      unittest.TestCase			):


	@cached_property
	def    _a     (self							):
				return ViTImageProcessor.from_pretrained("""google/vit-base-patch16-224"""							) if is_vision_available() else None


	@slow
	def    _a     (self							):
				A_     :						Optional[int]     =					ViTForImageClassification.from_pretrained("""google/vit-base-patch16-224"""							).to(lowercase							)

				A_     :						List[str]     =					self.default_image_processor
				A_     :						Tuple     =					prepare_img()
				A_     :						int     =					image_processor(images=lowercase	,				return_tensors="""pt"""							).to(lowercase							)

				# forward pass
				with torch.no_grad():
							A_     :						str     =					model(**lowercase							)

				# verify the logits
				A_     :						Dict     =					torch.Size((1, 1000)							)
				self.assertEqual(outputs.logits.shape	,				lowercase							)

				A_     :						str     =					torch.tensor([-0.27_44, 0.82_15, -0.08_36]							).to(lowercase							)

				self.assertTrue(torch.allclose(outputs.logits[0, :3]	,				lowercase	,				atol=1E-4							)							)


	@slow
	def    _a     (self							):
				# ViT models have an `interpolate_pos_encoding` argument in their forward method,
				# allowing to interpolate the pre-trained position embeddings in order to use
				# the model on higher resolutions. The DINO model by Facebook AI leverages this
				# to visualize self-attention on higher resolution images.
				A_     :						Optional[int]     =					ViTModel.from_pretrained("""facebook/dino-vits8"""							).to(lowercase							)

				A_     :						List[Any]     =					ViTImageProcessor.from_pretrained("""facebook/dino-vits8"""	,				size=480							)
				A_     :						Dict     =					prepare_img()
				A_     :						str     =					image_processor(images=lowercase	,				return_tensors="""pt"""							)
				A_     :						int     =					inputs.pixel_values.to(lowercase							)

				# forward pass
				with torch.no_grad():
							A_     :						int     =					model(lowercase	,				interpolate_pos_encoding=lowercase							)

				# verify the logits
				A_     :						int     =					torch.Size((1, 3601, 384)							)
				self.assertEqual(outputs.last_hidden_state.shape	,				lowercase							)

				A_     :						List[Any]     =					torch.tensor(
				    [[4.23_40, 4.39_06, -6.66_92], [4.54_63, 1.89_28, -6.72_57], [4.44_29, 0.84_96, -5.85_85]]							).to(lowercase							)

				self.assertTrue(torch.allclose(outputs.last_hidden_state[0, :3, :3]	,				lowercase	,				atol=1E-4							)							)


	@slow
	@require_accelerate
	@require_torch_gpu
	def    _a     (self							):
				A_     :						List[Any]     =					ViTModel.from_pretrained("""facebook/dino-vits8"""	,				torch_dtype=torch.floataa	,				device_map="""auto"""							)
				A_     :						int     =					self.default_image_processor

				A_     :						Any     =					prepare_img()
				A_     :						List[str]     =					image_processor(images=lowercase	,				return_tensors="""pt"""							)
				A_     :						Any     =					inputs.pixel_values.to(lowercase							)

				# forward pass to make sure inference works in fp16
				with torch.no_grad():
							A_     :						Optional[Any]     =					model(lowercase							)