




'''simple docstring'''






import argparse

from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection

from diffusers import UnCLIPImageVariationPipeline, UnCLIPPipeline


if __name__ == "__main__":
      A_			      =		argparse.ArgumentParser()

      parser.add_argument("--dump_path", default=None, type=str, required=True, help="Path to the output model.")

      parser.add_argument(
          "--txt2img_unclip",
          default="kakaobrain/karlo-v1-alpha",
          type=str,
          required=False,
          help="The pretrained txt2img unclip.",
      )

      A_			      =		parser.parse_args()

      A_			      =		UnCLIPPipeline.from_pretrained(args.txtaimg_unclip)

      A_			      =		CLIPImageProcessor()
      A_			      =		CLIPVisionModelWithProjection.from_pretrained("openai/clip-vit-large-patch14")

      A_			      =		UnCLIPImageVariationPipeline(
          decoder=txtaimg.decoder,
          text_encoder=txtaimg.text_encoder,
          tokenizer=txtaimg.tokenizer,
          text_proj=txtaimg.text_proj,
          feature_extractor=feature_extractor,
          image_encoder=image_encoder,
          super_res_first=txtaimg.super_res_first,
          super_res_last=txtaimg.super_res_last,
          decoder_scheduler=txtaimg.decoder_scheduler,
          super_res_scheduler=txtaimg.super_res_scheduler,
      )

      imgaimg.save_pretrained(args.dump_path)




'''simple docstring'''






def 							UpperCamelCase_     (       A__   :  bytes   ):
  '''simple docstring'''





  return "".join([hex(A__   )[2:].zfill(2   ).upper() for byte in list(A__   )]   )







def 							UpperCamelCase_     (       A__   :  str   ):
  '''simple docstring'''





  if (len(A__   ) % 2) != 0:
    raise ValueError(
        """Base16 encoded data is invalid:
Data does not have an even number of hex digits."""   )
  # Check the character set - the standard base16 alphabet
  # is uppercase according to RFC3548 section 6
  if not set(A__   ) <= set("""0123456789ABCDEF"""   ):
    raise ValueError(
        """Base16 encoded data is invalid:
Data is not uppercase hex or it contains invalid characters."""   )
  # For every two hexadecimal digits (= a byte), turn it into an integer.
  # Then, string the result together into bytes, and return it.
  return bytes(int(data[i] + data[i + 1]				,       16   ) for i in range(0				,       len(A__   )				,       2   )   )


if __name__ == "__main__":
  import doctest

  doctest.testmod()



