

'''simple docstring'''




import argparse
import json
from collections import OrderedDict

import torch
from huggingface_hub import cached_download, hf_hub_url

from transformers import AutoImageProcessor, CvtConfig, CvtForImageClassification




def 		UpperCamelCase			(					a					) ->      Dict:






     '''simple docstring'''
     __magic_name__		  =							[]
     embed.append(
         (
             F'''cvt.encoder.stages.{idx}.embedding.convolution_embeddings.projection.weight''',
             F'''stage{idx}.patch_embed.proj.weight''',
         )					)
     embed.append(
         (
             F'''cvt.encoder.stages.{idx}.embedding.convolution_embeddings.projection.bias''',
             F'''stage{idx}.patch_embed.proj.bias''',
         )					)
     embed.append(
         (
             F'''cvt.encoder.stages.{idx}.embedding.convolution_embeddings.normalization.weight''',
             F'''stage{idx}.patch_embed.norm.weight''',
         )					)
     embed.append(
         (
             F'''cvt.encoder.stages.{idx}.embedding.convolution_embeddings.normalization.bias''',
             F'''stage{idx}.patch_embed.norm.bias''',
         )					)
     return embed




def 		UpperCamelCase			(					a						,		a					) ->      Union[str, Any]:






     '''simple docstring'''
     __magic_name__		  =							[]
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_query.convolution_projection.convolution.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_q.conv.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_query.convolution_projection.normalization.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_q.bn.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_query.convolution_projection.normalization.bias''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_q.bn.bias''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_query.convolution_projection.normalization.running_mean''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_q.bn.running_mean''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_query.convolution_projection.normalization.running_var''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_q.bn.running_var''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_query.convolution_projection.normalization.num_batches_tracked''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_q.bn.num_batches_tracked''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_key.convolution_projection.convolution.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_k.conv.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_key.convolution_projection.normalization.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_k.bn.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_key.convolution_projection.normalization.bias''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_k.bn.bias''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_key.convolution_projection.normalization.running_mean''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_k.bn.running_mean''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_key.convolution_projection.normalization.running_var''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_k.bn.running_var''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_key.convolution_projection.normalization.num_batches_tracked''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_k.bn.num_batches_tracked''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_value.convolution_projection.convolution.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_v.conv.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_value.convolution_projection.normalization.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_v.bn.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_value.convolution_projection.normalization.bias''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_v.bn.bias''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_value.convolution_projection.normalization.running_mean''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_v.bn.running_mean''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_value.convolution_projection.normalization.running_var''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_v.bn.running_var''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.convolution_projection_value.convolution_projection.normalization.num_batches_tracked''',
             F'''stage{idx}.blocks.{cnt}.attn.conv_proj_v.bn.num_batches_tracked''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.projection_query.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.proj_q.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.projection_query.bias''',
             F'''stage{idx}.blocks.{cnt}.attn.proj_q.bias''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.projection_key.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.proj_k.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.projection_key.bias''',
             F'''stage{idx}.blocks.{cnt}.attn.proj_k.bias''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.projection_value.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.proj_v.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.attention.projection_value.bias''',
             F'''stage{idx}.blocks.{cnt}.attn.proj_v.bias''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.output.dense.weight''',
             F'''stage{idx}.blocks.{cnt}.attn.proj.weight''',
         )					)
     attention_weights.append(
         (
             F'''cvt.encoder.stages.{idx}.layers.{cnt}.attention.output.dense.bias''',
             F'''stage{idx}.blocks.{cnt}.attn.proj.bias''',
         )					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.intermediate.dense.weight''', F'''stage{idx}.blocks.{cnt}.mlp.fc1.weight''')					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.intermediate.dense.bias''', F'''stage{idx}.blocks.{cnt}.mlp.fc1.bias''')					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.output.dense.weight''', F'''stage{idx}.blocks.{cnt}.mlp.fc2.weight''')					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.output.dense.bias''', F'''stage{idx}.blocks.{cnt}.mlp.fc2.bias''')					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.layernorm_before.weight''', F'''stage{idx}.blocks.{cnt}.norm1.weight''')					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.layernorm_before.bias''', F'''stage{idx}.blocks.{cnt}.norm1.bias''')					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.layernorm_after.weight''', F'''stage{idx}.blocks.{cnt}.norm2.weight''')					)
     attention_weights.append(
         (F'''cvt.encoder.stages.{idx}.layers.{cnt}.layernorm_after.bias''', F'''stage{idx}.blocks.{cnt}.norm2.bias''')					)
     return attention_weights




def 		UpperCamelCase			(					a					) ->      Tuple:






     '''simple docstring'''
     __magic_name__		  =							[]
     token.append((F'''cvt.encoder.stages.{idx}.cls_token''', '''stage2.cls_token''')					)
     return token




def 		UpperCamelCase			(					) ->      List[str]:






     '''simple docstring'''
     __magic_name__		  =							[]
     head.append(('''layernorm.weight''', '''norm.weight''')					)
     head.append(('''layernorm.bias''', '''norm.bias''')					)
     head.append(('''classifier.weight''', '''head.weight''')					)
     head.append(('''classifier.bias''', '''head.bias''')					)
     return head




def 		UpperCamelCase			(					a						,		a						,		a						,		a					) ->      Dict:






     '''simple docstring'''
     __magic_name__		  =							'imagenet-1k-id2label.json'
     __magic_name__		  =							1000

     __magic_name__		  =							'huggingface/label-files'
     __magic_name__		  =							num_labels
     __magic_name__		  =							json.load(open(cached_download(hf_hub_url(__lowerCamelCase						,		__lowerCamelCase						,		repo_type='''dataset'''					)					)						,		'''r'''					)					)
     __magic_name__		  =							{int(__lowerCamelCase					): v for k, v in idalabel.items()}

     __magic_name__		  =							idalabel
     __magic_name__		  =							{v: k for k, v in idalabel.items()}

     __magic_name__		  =							CvtConfig(num_labels=__lowerCamelCase						,		idalabel=__lowerCamelCase						,		labelaid=__lowerCamelCase					)

     # For depth size 13 (13 = 1+2+10)
     if cvt_model.rsplit('''/'''						,		1					)[-1][4:6] == "13":
          __magic_name__		  =							[1, 2, 10]

     # For depth size 21 (21 = 1+4+16)
     elif cvt_model.rsplit('''/'''						,		1					)[-1][4:6] == "21":
          __magic_name__		  =							[1, 4, 16]

     # For wide cvt (similar to wide-resnet) depth size 24 (w24 = 2 + 2 20)
     else:
          __magic_name__		  =							[2, 2, 20]
          __magic_name__		  =							[3, 12, 16]
          __magic_name__		  =							[192, 768, 1024]

     __magic_name__		  =							CvtForImageClassification(__lowerCamelCase					)
     __magic_name__		  =							AutoImageProcessor.from_pretrained('''facebook/convnext-base-224-22k-1k'''					)
     __magic_name__		  =							image_size
     __magic_name__		  =							torch.load(__lowerCamelCase						,		map_location=torch.device('''cpu'''					)					)

     __magic_name__		  =							OrderedDict()
     __magic_name__		  =							[]

     for idx in range(len(config.depth					)					):
          if config.cls_token[idx]:
               __magic_name__		  =							list_of_state_dict + cls_token(__lowerCamelCase					)
          __magic_name__		  =							list_of_state_dict + embeddings(__lowerCamelCase					)
          for cnt in range(config.depth[idx]					):
               __magic_name__		  =							list_of_state_dict + attention(__lowerCamelCase						,		__lowerCamelCase					)

     __magic_name__		  =							list_of_state_dict + final()
     for gg in list_of_state_dict:
          print(__lowerCamelCase					)
     for i in range(len(__lowerCamelCase					)					):
          __magic_name__		  =							original_weights[list_of_state_dict[i][1]]

     model.load_state_dict(__lowerCamelCase					)
     model.save_pretrained(__lowerCamelCase					)
     image_processor.save_pretrained(__lowerCamelCase					)


# Download the weights from zoo: https://1drv.ms/u/s!AhIXJn_J-blW9RzF3rMW7SsLHa8h?e=blQ0Al

if __name__ == "__main__":
      _lowerCAmelCase            =		argparse.ArgumentParser()
      parser.add_argument(
          "--cvt_model",
          default="cvt-w24",
          type=str,
          help="Name of the cvt model you\'d like to convert.",
      )
      parser.add_argument(
          "--image_size",
          default=384,
          type=int,
          help="Input Image Size",
      )
      parser.add_argument(
          "--cvt_file_name",
          default=r"cvtmodels\CvT-w24-384x384-IN-22k.pth",
          type=str,
          help="Input Image Size",
      )
      parser.add_argument(
          "--pytorch_dump_folder_path", default=None, type=str, help="Path to the output PyTorch model directory."
      )

      _lowerCAmelCase            =		parser.parse_args()
      convert_cvt_checkpoint(args.cvt_model, args.image_size, args.cvt_file_name, args.pytorch_dump_folder_path)










'''simple docstring'''






import gc
import random
import unittest

import numpy as np
import torch
from PIL import Image
from transformers import XLMRobertaTokenizerFast

from diffusers import DDIMScheduler, KandinskyImgaImgPipeline, KandinskyPriorPipeline, UNetaDConditionModel, VQModel
from diffusers.pipelines.kandinsky.text_encoder import MCLIPConfig, MultilingualCLIP
from diffusers.utils import floats_tensor, load_image, load_numpy, slow, torch_device
from diffusers.utils.testing_utils import enable_full_determinism, require_torch_gpu

from ..test_pipelines_common import PipelineTesterMixin, assert_mean_pixel_difference


enable_full_determinism()






class    __magic_name__				(						lowerCAmelCase   ,unittest.TestCase							):
       UpperCAmelCase		 =KandinskyImgaImgPipeline
       UpperCAmelCase		 =["prompt", "image_embeds", "negative_image_embeds", "image"]
       UpperCAmelCase		 =[
           "prompt",
           "negative_prompt",
           "image_embeds",
           "negative_image_embeds",
           "image",
       ]
       UpperCAmelCase		 =[
           "generator",
           "height",
           "width",
           "strength",
           "guidance_scale",
           "negative_prompt",
           "num_inference_steps",
           "return_dict",
           "guidance_scale",
           "num_images_per_prompt",
           "output_type",
           "return_dict",
       ]
       UpperCAmelCase		 =False






       @property
       def      lowerCAmelCase						(     self)		->    str:




          '''simple docstring'''
          return 3_2






       @property
       def      lowerCAmelCase						(     self)		->    List[str]:




          '''simple docstring'''
          return 3_2






       @property
       def      lowerCAmelCase						(     self)		->    Union[str, Any]:




          '''simple docstring'''
          return self.time_input_dim






       @property
       def      lowerCAmelCase						(     self)		->    Optional[int]:




          '''simple docstring'''
          return self.time_input_dim * 4






       @property
       def      lowerCAmelCase						(     self)		->    Tuple:




          '''simple docstring'''
          return 1_0_0






       @property
       def      lowerCAmelCase						(     self)		->    Optional[Any]:




          '''simple docstring'''
          _UpperCAmelCase  :  int				       =XLMRobertaTokenizerFast.from_pretrained('YiYiXu/tiny-random-mclip-base')
          return tokenizer






       @property
       def      lowerCAmelCase						(     self)		->    Dict:




          '''simple docstring'''
          torch.manual_seed(0)
          _UpperCAmelCase  :  Optional[Any]				       =MCLIPConfig(
              numDims=self.cross_attention_dim    ,				transformerDimensions=self.text_embedder_hidden_size    ,				hidden_size=self.text_embedder_hidden_size    ,				intermediate_size=3_7    ,				num_attention_heads=4    ,				num_hidden_layers=5    ,				vocab_size=1_0_0_5    ,				)

          _UpperCAmelCase  :  List[Any]				       =MultilingualCLIP(snake_case)
          _UpperCAmelCase  :  List[Any]				       =text_encoder.eval()

          return text_encoder






       @property
       def      lowerCAmelCase						(     self)		->    Optional[Any]:




          '''simple docstring'''
          torch.manual_seed(0)

          _UpperCAmelCase  :  Union[str, Any]				       ={
              'in_channels': 4,
              # Out channels is double in channels because predicts mean and variance
              'out_channels': 8,
              'addition_embed_type': 'text_image',
              'down_block_types': ('ResnetDownsampleBlock2D', 'SimpleCrossAttnDownBlock2D'),
              'up_block_types': ('SimpleCrossAttnUpBlock2D', 'ResnetUpsampleBlock2D'),
              'mid_block_type': 'UNetMidBlock2DSimpleCrossAttn',
              'block_out_channels': (self.block_out_channels_a, self.block_out_channels_a * 2),
              'layers_per_block': 1,
              'encoder_hid_dim': self.text_embedder_hidden_size,
              'encoder_hid_dim_type': 'text_image_proj',
              'cross_attention_dim': self.cross_attention_dim,
              'attention_head_dim': 4,
              'resnet_time_scale_shift': 'scale_shift',
              'class_embed_type': None,
          }

          _UpperCAmelCase  :  Tuple				       =UNetaDConditionModel(**snake_case)
          return model






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




          '''simple docstring'''
          return {
              "block_out_channels": [3_2, 6_4],
              "down_block_types": ["DownEncoderBlock2D", "AttnDownEncoderBlock2D"],
              "in_channels": 3,
              "latent_channels": 4,
              "layers_per_block": 1,
              "norm_num_groups": 8,
              "norm_type": "spatial",
              "num_vq_embeddings": 1_2,
              "out_channels": 3,
              "up_block_types": [
                  "AttnUpDecoderBlock2D",
                  "UpDecoderBlock2D",
              ],
              "vq_embed_dim": 4,
          }






       @property
       def      lowerCAmelCase						(     self)		->    int:




          '''simple docstring'''
          torch.manual_seed(0)
          _UpperCAmelCase  :  List[Any]				       =VQModel(**self.dummy_movq_kwargs)
          return model






       def      lowerCAmelCase						(     self)		->    List[Any]:




          '''simple docstring'''
          _UpperCAmelCase  :  Tuple				       =self.dummy_text_encoder
          _UpperCAmelCase  :  Any				       =self.dummy_tokenizer
          _UpperCAmelCase  :  List[str]				       =self.dummy_unet
          _UpperCAmelCase  :  Union[str, Any]				       =self.dummy_movq

          _UpperCAmelCase  :  List[str]				       ={
              'num_train_timesteps': 1_0_0_0,
              'beta_schedule': 'linear',
              'beta_start': 0.0_00_85,
              'beta_end': 0.0_12,
              'clip_sample': False,
              'set_alpha_to_one': False,
              'steps_offset': 0,
              'prediction_type': 'epsilon',
              'thresholding': False,
          }

          _UpperCAmelCase  :  Dict				       =DDIMScheduler(**snake_case)

          _UpperCAmelCase  :  Optional[int]				       ={
              'text_encoder': text_encoder,
              'tokenizer': tokenizer,
              'unet': unet,
              'scheduler': scheduler,
              'movq': movq,
          }

          return components






       def      lowerCAmelCase						(     self    ,				snake_case    ,				snake_case=0)		->    int:




          '''simple docstring'''
          _UpperCAmelCase  :  List[Any]				       =floats_tensor((1, self.cross_attention_dim)    ,				rng=random.Random(snake_case)).to(snake_case)
          _UpperCAmelCase  :  Tuple				       =floats_tensor((1, self.cross_attention_dim)    ,				rng=random.Random(seed + 1)).to(snake_case)
          # create init_image
          _UpperCAmelCase  :  Dict				       =floats_tensor((1, 3, 6_4, 6_4)    ,				rng=random.Random(snake_case)).to(snake_case)
          _UpperCAmelCase  :  Optional[int]				       =image.cpu().permute(0    ,				2    ,				3    ,				1)[0]
          _UpperCAmelCase  :  List[Any]				       =Image.fromarray(np.uinta(snake_case)).convert('RGB').resize((2_5_6, 2_5_6))

          if str(snake_case).startswith('mps'):
             _UpperCAmelCase  :  str				       =torch.manual_seed(snake_case)
          else:
             _UpperCAmelCase  :  Optional[Any]				       =torch.Generator(device=snake_case).manual_seed(snake_case)
          _UpperCAmelCase  :  str				       ={
              'prompt': 'horse',
              'image': init_image,
              'image_embeds': image_embeds,
              'negative_image_embeds': negative_image_embeds,
              'generator': generator,
              'height': 6_4,
              'width': 6_4,
              'num_inference_steps': 1_0,
              'guidance_scale': 7.0,
              'strength': 0.2,
              'output_type': 'np',
          }
          return inputs






       def      lowerCAmelCase						(     self)		->    str:




          '''simple docstring'''
          _UpperCAmelCase  :  List[Any]				       ='cpu'

          _UpperCAmelCase  :  Dict				       =self.get_dummy_components()

          _UpperCAmelCase  :  Optional[Any]				       =self.pipeline_class(**snake_case)
          _UpperCAmelCase  :  List[str]				       =pipe.to(snake_case)

          pipe.set_progress_bar_config(disable=snake_case)

          _UpperCAmelCase  :  List[Any]				       =pipe(**self.get_dummy_inputs(snake_case))
          _UpperCAmelCase  :  List[str]				       =output.images

          _UpperCAmelCase  :  Any				       =pipe(
              **self.get_dummy_inputs(snake_case)    ,				return_dict=snake_case    ,				)[0]

          _UpperCAmelCase  :  Any				       =image[0, -3:, -3:, -1]
          _UpperCAmelCase  :  Tuple				       =image_from_tuple[0, -3:, -3:, -1]

          assert image.shape == (1, 6_4, 6_4, 3)

          _UpperCAmelCase  :  List[Any]				       =np.array(
              [0.61_47_49_43, 0.6_07_35_39, 0.43_30_85_44, 0.5_92_82_69, 0.47_49_35_95, 0.46_75_59_73, 0.4_61_38_38, 0.45_36_87_97, 0.50_11_92_33])
          assert (
              np.abs(image_slice.flatten() - expected_slice).max() < 1E-2
          ), f" expected_slice {expected_slice}, but got {image_slice.flatten()}"
          assert (
              np.abs(image_from_tuple_slice.flatten() - expected_slice).max() < 1E-2
          ), f" expected_slice {expected_slice}, but got {image_from_tuple_slice.flatten()}"






@slow
@require_torch_gpu
class    __magic_name__				(						unittest.TestCase							):






       def      lowerCAmelCase						(     self)		->    Optional[int]:




          '''simple docstring'''
          # clean up the VRAM after each test
          super().tearDown()
          gc.collect()
          torch.cuda.empty_cache()






       def      lowerCAmelCase						(     self)		->    Optional[int]:




          '''simple docstring'''
          _UpperCAmelCase  :  Tuple				       =load_numpy(
              'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'
              '/kandinsky/kandinsky_img2img_frog.npy')

          _UpperCAmelCase  :  List[Any]				       =load_image(
              'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main' '/kandinsky/cat.png')
          _UpperCAmelCase  :  Union[str, Any]				       ='A red cartoon frog, 4k'

          _UpperCAmelCase  :  Union[str, Any]				       =KandinskyPriorPipeline.from_pretrained(
              'kandinsky-community/kandinsky-2-1-prior'    ,				torch_dtype=torch.floataa)
          pipe_prior.to(snake_case)

          _UpperCAmelCase  :  Optional[Any]				       =KandinskyImgaImgPipeline.from_pretrained(
              'kandinsky-community/kandinsky-2-1'    ,				torch_dtype=torch.floataa)
          _UpperCAmelCase  :  Tuple				       =pipeline.to(snake_case)

          pipeline.set_progress_bar_config(disable=snake_case)

          _UpperCAmelCase  :  List[str]				       =torch.Generator(device='cpu').manual_seed(0)
          _UpperCAmelCase       ,			_UpperCAmelCase  :  List[Any]				       =pipe_prior(
              snake_case    ,				generator=snake_case    ,				num_inference_steps=5    ,				negative_prompt=''    ,				).to_tuple()

          _UpperCAmelCase  :  str				       =pipeline(
              snake_case    ,				image=snake_case    ,				image_embeds=snake_case    ,				negative_image_embeds=snake_case    ,				generator=snake_case    ,				num_inference_steps=1_0_0    ,				height=7_6_8    ,				width=7_6_8    ,				strength=0.2    ,				output_type='np'    ,				)

          _UpperCAmelCase  :  Any				       =output.images[0]

          assert image.shape == (7_6_8, 7_6_8, 3)

          assert_mean_pixel_difference(snake_case    ,				snake_case)

