

'''simple docstring'''

import argparse

from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection

from diffusers import UnCLIPImageVariationPipeline, UnCLIPPipeline


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

       UpperCAmelCase					 =	parser.parse_args()

       UpperCAmelCase					 =	UnCLIPPipeline.from_pretrained(args.txtaimg_unclip)

       UpperCAmelCase					 =	CLIPImageProcessor()
       UpperCAmelCase					 =	CLIPVisionModelWithProjection.from_pretrained('''openai/clip-vit-large-patch14''')

       UpperCAmelCase					 =	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				(   lowerCAmelCase__:					int = 1000   )      -> int:
							SCREAMING_SNAKE_CASE_				=				2**power
							SCREAMING_SNAKE_CASE_				=				str(lowerCAmelCase__   )
							SCREAMING_SNAKE_CASE_				=				list(lowerCAmelCase__   )
							SCREAMING_SNAKE_CASE_				=				0

							for i in list_num:
														sum_of_num += int(lowerCAmelCase__   )

							return sum_of_num


if __name__ == "__main__":
							SCREAMING_SNAKE_CASE							:				Tuple     	=			int(input("Enter the power of 2: ").strip())
							print("2 ^ ", power, " = ", 2**power)
							SCREAMING_SNAKE_CASE							:				str     	=			solution(power)
							print("Sum of the digits is: ", result)





