




'''simple docstring'''





from __future__ import annotations

import random
import string




class      UpperCamelCase       :



  def __init__(			self		,     __lowerCAmelCase = None ):







      '''simple docstring'''


      _a				:							List[Any]							       =      passcode or self.__passcode_creator()
      _a				:							List[str]							       =      self.__make_key_list()
      _a				:							Optional[int]							       =      self.__make_shift_key()



  def __str__(			self ):







      '''simple docstring'''


      return "".join(self.__passcode )



  def     a_							(			self		,     __lowerCAmelCase ):







      '''simple docstring'''


      for i in range(1		,     len(__lowerCAmelCase )		,     2 ):
          iterlist[i] *= -1
      return iterlist



  def     a_							(			self ):







      '''simple docstring'''


      _a				:							Optional[int]							       =      string.ascii_letters + string.digits
      _a				:							str							       =      [random.choice(__lowerCAmelCase ) for _ in range(random.randint(10		,     20 ) )]
      return password



  def     a_							(			self ):







      '''simple docstring'''


      _a				:							Dict							       =      (
          string.ascii_letters + string.digits + string.punctuation + """ \t\n"""
      )

      _a				:							str							       =      []

      # creates points known as breakpoints to break the key_list_options at those
      # points and pivot each substring
      _a				:							Dict							       =      sorted(set(self.__passcode ) )
      _a				:							list[str]							       =      []

      # algorithm for creating a new shuffled list, keys_l, out of key_list_options
      for i in key_list_options:
          temp_list.extend(__lowerCAmelCase )

          # checking breakpoints at which to pivot temporary sublist and add it into
          # keys_l
          if i in breakpoints or i == key_list_options[-1]:
              keys_l.extend(temp_list[::-1] )
              temp_list.clear()

        # returning a shuffled keys_l to prevent brute force guessing of shift key
      return keys_l



  def     a_							(			self ):







      '''simple docstring'''


      _a				:							Optional[Any]							       =      sum(self.__neg_pos([ord(__lowerCAmelCase ) for x in self.__passcode] ) )
      return num if num > 0 else len(self.__passcode )



  def     a_							(			self		,     __lowerCAmelCase ):







      '''simple docstring'''


      _a				:							Tuple							       =      """"""

      # decoding shift like Caesar cipher algorithm implementing negative shift or
      # reverse shift or left shift
      for i in encoded_message:
          _a				:							List[str]							       =      self.__key_list.index(__lowerCAmelCase )
          decoded_message += self.__key_list[
              (position - self.__shift_key) % -len(self.__key_list )
          ]

      return decoded_message



  def     a_							(			self		,     __lowerCAmelCase ):







      '''simple docstring'''


      _a				:							List[str]							       =      """"""

      # encoding shift like Caesar cipher algorithm implementing positive shift or
      # forward shift or right shift
      for i in plaintext:
          _a				:							Any							       =      self.__key_list.index(__lowerCAmelCase )
          encoded_message += self.__key_list[
              (position + self.__shift_key) % len(self.__key_list )
          ]

      return encoded_message






def UpperCAmelCase    (       UpperCAmelCase_ = "Hello, this is a modified Caesar cipher"						):
    _a				:							Union[str, Any]							       =      ShuffledShiftCipher()
    return cipa.decrypt(cipa.encrypt(UpperCAmelCase_						)						)


if __name__ == "__main__":
  import doctest

  doctest.testmod()



"""simple docstring"""


import argparse

import torch
from clip import load

from transformers import CLIPConfig, CLIPModel







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> Optional[Any]:
     snake_case_			,      snake_case_			,      snake_case_   			=   pt_attn_layer.in_proj_weight.chunk(3		,   dim=0   )
     snake_case_			,      snake_case_			,      snake_case_   			=   pt_attn_layer.in_proj_bias.chunk(3		,   dim=0   )

     snake_case_   			=   pt_attn_layer.out_proj.weight
     snake_case_   			=   pt_attn_layer.out_proj.bias

     snake_case_   			=   q_proj
     snake_case_   			=   q_proj_bias

     snake_case_   			=   k_proj
     snake_case_   			=   k_proj_bias

     snake_case_   			=   v_proj
     snake_case_   			=   v_proj_bias

     snake_case_   			=   out_proj_weights
     snake_case_   			=   out_proj_bias







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> int:
     copy_linear(hf_mlp.fca		,   pt_mlp.c_fc   )
     copy_linear(hf_mlp.fca		,   pt_mlp.c_proj   )







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> int:
     snake_case_   			=   pt_linear.weight
     snake_case_   			=   pt_linear.bias







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> str:
     # copy layer norms
     copy_linear(hf_layer.layer_norma		,   pt_layer.ln_a   )
     copy_linear(hf_layer.layer_norma		,   pt_layer.ln_a   )

     # copy MLP
     copy_mlp(hf_layer.mlp		,   pt_layer.mlp   )

     # copy attn
     copy_attn_layer(hf_layer.self_attn		,   pt_layer.attn   )







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> Optional[Any]:
     for hf_layer, pt_layer in zip(UpperCAmelCase		,   UpperCAmelCase   ):
          copy_layer(UpperCAmelCase		,   UpperCAmelCase   )







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> Tuple:
     # copy  embeds
     snake_case_   			=   pt_model.token_embedding.weight
     snake_case_   			=   pt_model.positional_embedding

     # copy layer norm
     copy_linear(hf_encoder.final_layer_norm		,   pt_model.ln_final   )

     # copy hidden layers
     copy_layers(hf_encoder.encoder.layers		,   pt_model.transformer.resblocks   )







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> int:
     # copy projection
     snake_case_   			=   pt_model.text_projection.data.T

     # copy text encoder
     copy_encoder(hf_model.text_model		,   UpperCAmelCase   )







def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase   )							-> List[str]:
     # copy projection
     snake_case_   			=   pt_model.visual.proj.data.T

     # copy layer norms
     copy_linear(hf_model.vision_model.pre_layrnorm		,   pt_model.visual.ln_pre   )
     copy_linear(hf_model.vision_model.post_layernorm		,   pt_model.visual.ln_post   )

     # copy embeds
     snake_case_   			=   pt_model.visual.conva.weight.data
     snake_case_   			=   pt_model.visual.class_embedding
     snake_case_   			=   pt_model.visual.positional_embedding.data

     # copy encoder
     copy_layers(hf_model.vision_model.encoder.layers		,   pt_model.visual.transformer.resblocks   )







@torch.no_grad()
def    UpperCAmelCase       (						UpperCAmelCase		,   UpperCAmelCase		,   UpperCAmelCase=None   )							-> Optional[Any]:
     if config_path is not None:
          snake_case_   			=   CLIPConfig.from_pretrained(UpperCAmelCase   )
     else:
          snake_case_   			=   CLIPConfig(projection_dim=512		,   text_config={}		,   vision_config={}   )

     snake_case_   			=   CLIPModel(UpperCAmelCase   ).eval()

     snake_case_			,      snake_case_   			=   load(UpperCAmelCase		,   device='cpu'		,   jit=UpperCAmelCase   )
     snake_case_   			=   pt_model.eval()

     copy_text_model_and_projection(UpperCAmelCase		,   UpperCAmelCase   )
     copy_vison_model_and_projection(UpperCAmelCase		,   UpperCAmelCase   )
     snake_case_   			=   pt_model.logit_scale

     snake_case_   			=   torch.arange(0		,   77   ).unsqueeze(0   )
     snake_case_   			=   torch.randn(1		,   3		,   224		,   224   )

     snake_case_   			=   hf_model(input_ids=UpperCAmelCase		,   pixel_values=UpperCAmelCase		,   return_dict=UpperCAmelCase   )
     snake_case_   			=   hf_outputs.logits_per_image
     snake_case_   			=   hf_outputs.logits_per_text
     snake_case_			,      snake_case_   			=   pt_model(UpperCAmelCase		,   UpperCAmelCase   )

     assert torch.allclose(UpperCAmelCase		,   UpperCAmelCase		,   atol=1e-3   )
     assert torch.allclose(UpperCAmelCase		,   UpperCAmelCase		,   atol=1e-3   )

     hf_model.save_pretrained(UpperCAmelCase   )


if __name__ == "__main__":
    __UpperCamelCase										=     argparse.ArgumentParser()
    parser.add_argument('''--pytorch_dump_folder_path''', default=None, type=str, help='''Path to the output PyTorch model.''')
    parser.add_argument('''--checkpoint_path''', default=None, type=str, help='''Path to fairseq checkpoint''')
    parser.add_argument('''--config_path''', default=None, type=str, help='''Path to hf config.json of model to convert''')
    __UpperCamelCase										=     parser.parse_args()

    convert_clip_checkpoint(args.checkpoint_path, args.pytorch_dump_folder_path, args.config_path)

