



'''simple docstring'''






from __future__ import annotations

import typing
from collections.abc import Iterable

import numpy as np

lowerCAmelCase: Tuple							=						typing.Union[Iterable[float], Iterable[int], np.ndarray]  # noqa: UP007
lowerCAmelCase: int							=						typing.Union[np.floataa, int, float]  # noqa: UP007
def 				lowerCamelCase__	(						_A			,      _A    ):
  return np.sqrt(np.sum((np.asarray(_A    ) - np.asarray(_A    )) ** 2    )    )
def 				lowerCamelCase__	(						_A			,      _A    ):
  return sum((va - va) ** 2 for va, va in zip(_A			,      _A    )    ) ** (1 / 2)


if __name__ == "__main__":
     def 				lowerCamelCase__	(						):
       from timeit import timeit

       print('Without Numpy'    )
       print(
           timeit(
               'euclidean_distance_no_np([1, 2, 3], [4, 5, 6])'			,      number=1_0000			,      globals=globals()			,      )    )
       print('With Numpy'    )
       print(
           timeit(
               'euclidean_distance([1, 2, 3], [4, 5, 6])'			,      number=1_0000			,      globals=globals()			,      )    )

     benchmark()




'''simple docstring'''






import argparse

from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection

from diffusers import UnCLIPImageVariationPipeline, UnCLIPPipeline


if __name__ == "__main__":
     lowerCAmelCase: Any							=						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.',
     )

     lowerCAmelCase: Optional[int]							=						parser.parse_args()

     lowerCAmelCase: List[Any]							=						UnCLIPPipeline.from_pretrained(args.txtaimg_unclip)

     lowerCAmelCase: Optional[Any]							=						CLIPImageProcessor()
     lowerCAmelCase: Tuple							=						CLIPVisionModelWithProjection.from_pretrained('openai/clip-vit-large-patch14')

     lowerCAmelCase: List[str]							=						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)