






"""simple docstring"""



import unittest

from transformers import load_tool

from .test_tools_common import ToolTesterMixin


__A		:       Dict  				=		'''
Hugging Face was founded in 2016 by French entrepreneurs Clément Delangue, Julien Chaumond, and Thomas Wolf originally as a company that developed a chatbot app targeted at teenagers.[2] After open-sourcing the model behind the chatbot, the company pivoted to focus on being a platform for machine learning.

In March 2021, Hugging Face raised $40 million in a Series B funding round.[3]

On April 28, 2021, the company launched the BigScience Research Workshop in collaboration with several other research groups to release an open large language model.[4] In 2022, the workshop concluded with the announcement of BLOOM, a multilingual large language model with 176 billion parameters.[5]
'''





class 					_UpperCAmelCase       ( unittest.TestCase   ,						_A							):




					def   A  (				self     :						List[Any]   )  ->  Dict:
							lowercase_ : Optional[int]							       =   load_tool('''text-question-answering'''   )
							self.tool.setup()
							lowercase_ : Union[str, Any]							       =   load_tool('''text-question-answering'''    ,						remote=A   )




					def   A  (				self     :						Any   )  ->  List[str]:
							lowercase_ : Union[str, Any]							       =   self.tool(A    ,						'''What did Hugging Face do in April 2021?'''   )
							self.assertEqual(A    ,						'''launched the BigScience Research Workshop'''   )




					def   A  (				self     :						str   )  ->  List[str]:
							lowercase_ : int							       =   self.remote_tool(A    ,						'''What did Hugging Face do in April 2021?'''   )
							self.assertEqual(A    ,						'''launched the BigScience Research Workshop'''   )




					def   A  (				self     :						List[Any]   )  ->  int:
							lowercase_ : Optional[Any]							       =   self.tool(text=A    ,						question='''What did Hugging Face do in April 2021?'''   )
							self.assertEqual(A    ,						'''launched the BigScience Research Workshop'''   )






					def   A  (				self     :						List[str]   )  ->  Optional[int]:
							lowercase_ : int							       =   self.remote_tool(text=A    ,						question='''What did Hugging Face do in April 2021?'''   )
							self.assertEqual(A    ,						'''launched the BigScience Research Workshop'''   )











"""simple docstring"""



import argparse

from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection

from diffusers import UnCLIPImageVariationPipeline, UnCLIPPipeline


if __name__ == "__main__":
						__A		:       str  				=		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		:       str  				=		parser.parse_args()

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

						__A		:       Dict  				=		CLIPImageProcessor()
						__A		:       Union[str, Any]  				=		CLIPVisionModelWithProjection.from_pretrained('''openai/clip-vit-large-patch14''')

						__A		:       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)



