



"""simple docstring"""





import unittest

import torch

from diffusers import VQModel
from diffusers.utils import floats_tensor, torch_device
from diffusers.utils.testing_utils import enable_full_determinism

from .test_modeling_common import ModelTesterMixin, UNetTesterMixin


enable_full_determinism()
class  lowercase__  (       SCREAMING_SNAKE_CASE							,   SCREAMING_SNAKE_CASE							,   unittest.TestCase ):







       '''simple docstring'''


       UpperCamelCase		=	VQModel
       UpperCamelCase		=	'''sample'''



       @property
       def 	lowercase__							(				self     :    str						,							_UpperCAmelCase     :    str=(32, 32)       )					->		Union[str, Any]:


             '''simple docstring'''




             UpperCAmelCase_     					=				4
             UpperCAmelCase_     					=				3

             UpperCAmelCase_     					=				floats_tensor((batch_size, num_channels) + sizes       ).to(_UpperCAmelCase       )

             return {"sample": image}



       @property
       def 	lowercase__							(				self     :    Dict       )					->		List[Any]:


             '''simple docstring'''




             return (3, 32, 32)



       @property
       def 	lowercase__							(				self     :    int       )					->		Dict:


             '''simple docstring'''




             return (3, 32, 32)



       def 	lowercase__							(				self     :    Union[str, Any]       )					->		Dict:


             '''simple docstring'''




             UpperCAmelCase_     					=				{
                 "block_out_channels": [32, 64],
                 "in_channels": 3,
                 "out_channels": 3,
                 "down_block_types": ["DownEncoderBlock2D", "DownEncoderBlock2D"],
                 "up_block_types": ["UpDecoderBlock2D", "UpDecoderBlock2D"],
                 "latent_channels": 3,
             }
             UpperCAmelCase_     					=				self.dummy_input
             return init_dict, inputs_dict



       def 	lowercase__							(				self     :    str       )					->		List[str]:


             '''simple docstring'''




             pass



       def 	lowercase__							(				self     :    Any       )					->		Optional[int]:


             '''simple docstring'''




             pass



       def 	lowercase__							(				self     :    Tuple       )					->		Any:


             '''simple docstring'''




             UpperCAmelCase_					,       UpperCAmelCase_     					=				VQModel.from_pretrained("fusing/vqgan-dummy"						,							output_loading_info=_UpperCAmelCase       )
             self.assertIsNotNone(_UpperCAmelCase       )
             self.assertEqual(len(loading_info["missing_keys"]       )						,							0       )

             model.to(_UpperCAmelCase       )
             UpperCAmelCase_     					=				model(**self.dummy_input       )

             assert image is not None, "Make sure output is not None"







       def 	lowercase__							(				self     :    str       )					->		List[str]:


             '''simple docstring'''




             UpperCAmelCase_     					=				VQModel.from_pretrained("fusing/vqgan-dummy"       )
             model.to(_UpperCAmelCase       ).eval()

             torch.manual_seed(0       )
             if torch.cuda.is_available():
                   torch.cuda.manual_seed_all(0       )

             UpperCAmelCase_     					=				torch.randn(1						,							model.config.in_channels						,							model.config.sample_size						,							model.config.sample_size       )
             UpperCAmelCase_     					=				image.to(_UpperCAmelCase       )
             with torch.no_grad():
                   UpperCAmelCase_     					=				model(_UpperCAmelCase       ).sample

             UpperCAmelCase_     					=				output[0, -1, -3:, -3:].flatten().cpu()
             # fmt: off
             UpperCAmelCase_     					=				torch.tensor([-0.0153, -0.4044, -0.1880, -0.5161, -0.2418, -0.4072, -0.1612, -0.0633, -0.0143]       )
             # fmt: on
             self.assertTrue(torch.allclose(_UpperCAmelCase						,							_UpperCAmelCase						,							atol=1e-3       )       )











import json
import os
import shutil
import tempfile
import unittest

import numpy as np

from transformers import BertTokenizerFast
from transformers.models.bert.tokenization_bert import VOCAB_FILES_NAMES, BertTokenizer
from transformers.testing_utils import require_tokenizers, require_vision
from transformers.utils import IMAGE_PROCESSOR_NAME, is_vision_available


if is_vision_available():
    from PIL import Image

    from transformers import VisionTextDualEncoderProcessor, ViTImageProcessor

@require_tokenizers
@require_vision
class   lowerCAmelCase				(   unittest.TestCase     ):
 def 	A_      ( self    : str      )     ->      Dict:
     lowerCamelCase__       : Optional[Any]   					=						tempfile.mkdtemp()

     # fmt: off
     lowerCamelCase__       : Tuple   					=						['[UNK]', '[CLS]', '[SEP]', '[PAD]', '[MASK]', 'want', '##want', '##ed', 'wa', 'un', 'runn', '##ing', ',', 'low', 'lowest']
     # fmt: on
     lowerCamelCase__       : int   					=						os.path.join(self.tmpdirname     ,	VOCAB_FILES_NAMES['vocab_file']      )
     with open(self.vocab_file     ,	'w'     ,	encoding='utf-8'      ) as vocab_writer:
         vocab_writer.write(''.join([x + '\n' for x in vocab_tokens]      )      )

     lowerCamelCase__       : Any   					=						{
         'do_resize': True,
         'size': {'height': 18, 'width': 18},
         'do_normalize': True,
         'image_mean': [0.5, 0.5, 0.5],
         'image_std': [0.5, 0.5, 0.5],
     }
     lowerCamelCase__       : int   					=						os.path.join(self.tmpdirname     ,	UpperCAmelCase      )
     with open(self.image_processor_file     ,	'w'     ,	encoding='utf-8'      ) as fp:
         json.dump(UpperCAmelCase     ,	UpperCAmelCase      )
 def 	A_      ( self    : List[str]     ,	**UpperCAmelCase    : str      )     ->      int:
     return BertTokenizer.from_pretrained(self.tmpdirname     ,	**UpperCAmelCase      )
 def 	A_      ( self    : Optional[int]     ,	**UpperCAmelCase    : List[str]      )     ->      List[str]:
     return ViTImageProcessor.from_pretrained(self.tmpdirname     ,	**UpperCAmelCase      )
 def 	A_      ( self    : int      )     ->      Any:
     shutil.rmtree(self.tmpdirname      )
 def 	A_      ( self    : str      )     ->      str:
     lowerCamelCase__       : List[Any]   					=						[np.random.randint(255     ,	size=(3, 30, 400)     ,	dtype=np.uinta      )]

     lowerCamelCase__       : Union[str, Any]   					=						[Image.fromarray(np.moveaxis(UpperCAmelCase     ,	0     ,	-1      )      ) for x in image_inputs]

     return image_inputs
 def 	A_      ( self    : Dict      )     ->      Dict:
     lowerCamelCase__       : List[Any]   					=						self.get_tokenizer()
     lowerCamelCase__       : Tuple   					=						self.get_image_processor()

     lowerCamelCase__       : Union[str, Any]   					=						VisionTextDualEncoderProcessor(tokenizer=UpperCAmelCase     ,	image_processor=UpperCAmelCase      )

     processor.save_pretrained(self.tmpdirname      )
     lowerCamelCase__       : List[str]   					=						VisionTextDualEncoderProcessor.from_pretrained(self.tmpdirname      )

     self.assertEqual(processor.tokenizer.get_vocab()     ,	tokenizer.get_vocab()      )
     self.assertIsInstance(processor.tokenizer     ,	(BertTokenizer, BertTokenizerFast)      )

     self.assertEqual(processor.image_processor.to_json_string()     ,	image_processor.to_json_string()      )
     self.assertIsInstance(processor.image_processor     ,	UpperCAmelCase      )
 def 	A_      ( self    : List[str]      )     ->      Tuple:
     lowerCamelCase__       : List[Any]   					=						VisionTextDualEncoderProcessor(
         tokenizer=self.get_tokenizer()     ,	image_processor=self.get_image_processor()      )
     processor.save_pretrained(self.tmpdirname      )

     lowerCamelCase__       : Tuple   					=						self.get_tokenizer(bos_token='(BOS)'     ,	eos_token='(EOS)'      )
     lowerCamelCase__       : int   					=						self.get_image_processor(do_normalize=UpperCAmelCase     ,	padding_value=1.0      )

     lowerCamelCase__       : Union[str, Any]   					=						VisionTextDualEncoderProcessor.from_pretrained(
         self.tmpdirname     ,	bos_token='(BOS)'     ,	eos_token='(EOS)'     ,	do_normalize=UpperCAmelCase     ,	padding_value=1.0      )

     self.assertEqual(processor.tokenizer.get_vocab()     ,	tokenizer_add_kwargs.get_vocab()      )
     self.assertIsInstance(processor.tokenizer     ,	(BertTokenizer, BertTokenizerFast)      )

     self.assertEqual(processor.image_processor.to_json_string()     ,	image_processor_add_kwargs.to_json_string()      )
     self.assertIsInstance(processor.image_processor     ,	UpperCAmelCase      )
 def 	A_      ( self    : Any      )     ->      Tuple:
     lowerCamelCase__       : List[Any]   					=						self.get_image_processor()
     lowerCamelCase__       : Any   					=						self.get_tokenizer()

     lowerCamelCase__       : Optional[int]   					=						VisionTextDualEncoderProcessor(tokenizer=UpperCAmelCase     ,	image_processor=UpperCAmelCase      )

     lowerCamelCase__       : Optional[int]   					=						self.prepare_image_inputs()

     lowerCamelCase__       : int   					=						image_processor(UpperCAmelCase     ,	return_tensors='np'      )
     lowerCamelCase__       : List[str]   					=						processor(images=UpperCAmelCase     ,	return_tensors='np'      )

     for key in input_feat_extract.keys():
         self.assertAlmostEqual(input_feat_extract[key].sum()     ,	input_processor[key].sum()     ,	delta=1e-2      )
 def 	A_      ( self    : Tuple      )     ->      List[Any]:
     lowerCamelCase__       : Dict   					=						self.get_image_processor()
     lowerCamelCase__       : List[str]   					=						self.get_tokenizer()

     lowerCamelCase__       : str   					=						VisionTextDualEncoderProcessor(tokenizer=UpperCAmelCase     ,	image_processor=UpperCAmelCase      )

     lowerCamelCase__       : Tuple   					=						'lower newer'

     lowerCamelCase__       : Tuple   					=						processor(text=UpperCAmelCase      )

     lowerCamelCase__       : Any   					=						tokenizer(UpperCAmelCase      )

     for key in encoded_tok.keys():
         self.assertListEqual(encoded_tok[key]     ,	encoded_processor[key]      )
 def 	A_      ( self    : Optional[int]      )     ->      Any:
     lowerCamelCase__       : Optional[int]   					=						self.get_image_processor()
     lowerCamelCase__       : Tuple   					=						self.get_tokenizer()

     lowerCamelCase__       : Tuple   					=						VisionTextDualEncoderProcessor(tokenizer=UpperCAmelCase     ,	image_processor=UpperCAmelCase      )

     lowerCamelCase__       : str   					=						'lower newer'
     lowerCamelCase__       : Dict   					=						self.prepare_image_inputs()

     lowerCamelCase__       : int   					=						processor(text=UpperCAmelCase     ,	images=UpperCAmelCase      )

     self.assertListEqual(list(inputs.keys()      )     ,	['input_ids', 'token_type_ids', 'attention_mask', 'pixel_values']      )

     # test if it raises when no input is passed
     with self.assertRaises(UpperCAmelCase      ):
         processor()
 def 	A_      ( self    : Optional[Any]      )     ->      List[str]:
     lowerCamelCase__       : Dict   					=						self.get_image_processor()
     lowerCamelCase__       : Union[str, Any]   					=						self.get_tokenizer()

     lowerCamelCase__       : List[str]   					=						VisionTextDualEncoderProcessor(tokenizer=UpperCAmelCase     ,	image_processor=UpperCAmelCase      )

     lowerCamelCase__       : List[str]   					=						[[1, 4, 5, 8, 1, 0, 8], [3, 4, 3, 1, 1, 8, 9]]

     lowerCamelCase__       : Dict   					=						processor.batch_decode(UpperCAmelCase      )
     lowerCamelCase__       : List[str]   					=						tokenizer.batch_decode(UpperCAmelCase      )

     self.assertListEqual(UpperCAmelCase     ,	UpperCAmelCase      )







 def 	A_      ( self    : Optional[int]      )     ->      List[str]:
     lowerCamelCase__       : Tuple   					=						self.get_image_processor()
     lowerCamelCase__       : List[str]   					=						self.get_tokenizer()

     lowerCamelCase__       : Union[str, Any]   					=						VisionTextDualEncoderProcessor(tokenizer=UpperCAmelCase     ,	image_processor=UpperCAmelCase      )

     lowerCamelCase__       : Dict   					=						'lower newer'
     lowerCamelCase__       : List[Any]   					=						self.prepare_image_inputs()

     lowerCamelCase__       : Union[str, Any]   					=						processor(text=UpperCAmelCase     ,	images=UpperCAmelCase      )

     self.assertListEqual(list(inputs.keys()      )     ,	processor.model_input_names      )





