
'''simple docstring'''



import random
import unittest

import torch

from diffusers import IFImgaImgPipeline
from diffusers.utils import floats_tensor
from diffusers.utils.import_utils import is_xformers_available
from diffusers.utils.testing_utils import skip_mps, torch_device

from ..pipeline_params import (
    TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS,
    TEXT_GUIDED_IMAGE_VARIATION_PARAMS,
)
from ..test_pipelines_common import PipelineTesterMixin
from . import IFPipelineTesterMixin

@skip_mps
class 				_a			(_lowerCamelCase					,   _lowerCamelCase					,   unittest.TestCase):





       """simple docstring"""




       SCREAMING_SNAKE_CASE =					IFImgaImgPipeline
       SCREAMING_SNAKE_CASE =					TEXT_GUIDED_IMAGE_VARIATION_PARAMS - {'width', 'height'}
       SCREAMING_SNAKE_CASE =					TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS
       SCREAMING_SNAKE_CASE =					PipelineTesterMixin.required_optional_params - {'latents'}






       def     UpperCamelCase			(							self )      ->     Any:
          return self._get_dummy_components()






       def     UpperCamelCase			(							self    ,							A__    ,							A__=0 )      ->     Optional[Any]:
          if str(A__ ).startswith("""mps""" ):
             _SCREAMING_SNAKE_CASE       	=			torch.manual_seed(A__ )
          else:
             _SCREAMING_SNAKE_CASE       	=			torch.Generator(device=A__ ).manual_seed(A__ )

          _SCREAMING_SNAKE_CASE       	=			floats_tensor((1, 3, 32, 32)    ,							rng=random.Random(A__ ) ).to(A__ )

          _SCREAMING_SNAKE_CASE       	=			{
              """prompt""": """A painting of a squirrel eating a burger""",
              """image""": image,
              """generator""": generator,
              """num_inference_steps""": 2,
              """output_type""": """numpy""",
          }

          return inputs






       def     UpperCamelCase			(							self )      ->     Dict:
          self._test_save_load_optional_components()






       @unittest.skipIf(
           torch_device != """cuda""" or not is_xformers_available()    ,							reason="""XFormers attention is only available with CUDA and `xformers` installed"""    ,							)
       def     UpperCamelCase			(							self )      ->     int:
          self._test_xformers_attention_forwardGenerator_pass(expected_max_diff=1E-3 )






       @unittest.skipIf(torch_device != """cuda"""    ,							reason="""float16 requires CUDA""" )
       def     UpperCamelCase			(							self )      ->     Optional[int]:
          # Due to non-determinism in save load of the hf-internal-testing/tiny-random-t5 text encoder
          super().test_save_load_floataa(expected_max_diff=1E-1 )






       @unittest.skipIf(torch_device != """cuda"""    ,							reason="""float16 requires CUDA""" )
       def     UpperCamelCase			(							self )      ->     int:
          super().test_floataa_inference(expected_max_diff=1E-1 )






       def     UpperCamelCase			(							self )      ->     Optional[Any]:
          self._test_attention_slicing_forward_pass(expected_max_diff=1E-2 )






       def     UpperCamelCase			(							self )      ->     int:
          self._test_save_load_local()






       def     UpperCamelCase			(							self )      ->     Any:
          self._test_inference_batch_single_identical(
              expected_max_diff=1E-2    ,							)








'''simple docstring'''



#
# Sort large text files in a minimum amount of memory
#
import argparse
import os

class 				_a			:





       """simple docstring"""




       SCREAMING_SNAKE_CASE =					'block_{0}.dat'






       def __init__(							self    ,							A__ )      ->     Union[str, Any]:
          _SCREAMING_SNAKE_CASE       	=			filename
          _SCREAMING_SNAKE_CASE       	=			[]






       def     UpperCamelCase			(							self    ,							A__    ,							A__ )      ->     Optional[Any]:
          _SCREAMING_SNAKE_CASE       	=			self.BLOCK_FILENAME_FORMAT.format(A__ )
          with open(A__    ,							"""w""" ) as file:
             file.write(A__ )
          self.block_filenames.append(A__ )






       def     UpperCamelCase			(							self )      ->     Tuple:
          return self.block_filenames






       def     UpperCamelCase			(							self    ,							A__    ,							A__=None )      ->     Any:
          _SCREAMING_SNAKE_CASE       	=			0
          with open(self.filename ) as file:
             while True:
                _SCREAMING_SNAKE_CASE       	=			file.readlines(A__ )

                if lines == []:
                   break

                if sort_key is None:
                   lines.sort()
                else:
                   lines.sort(key=A__ )

                self.write_block("""""".join(A__ )    ,							A__ )
                i += 1






       def     UpperCamelCase			(							self )      ->     List[Any]:
          map(os.remove    ,							self.block_filenames )






class 				_a			:





       """simple docstring"""




       def     UpperCamelCase			(							self    ,							A__ )      ->     str:
          _SCREAMING_SNAKE_CASE       	=			-1
          _SCREAMING_SNAKE_CASE       	=			None

          for i in range(len(A__ ) ):
             if min_str is None or choices[i] < min_str:
                _SCREAMING_SNAKE_CASE       	=			i

          return min_index






class 				_a			:





       """simple docstring"""




       def __init__(							self    ,							A__ )      ->     Dict:
          _SCREAMING_SNAKE_CASE       	=			files
          _SCREAMING_SNAKE_CASE       	=			set()
          _SCREAMING_SNAKE_CASE       	=			len(A__ )
          _SCREAMING_SNAKE_CASE       	=			{i: None for i in range(self.num_buffers )}






       def     UpperCamelCase			(							self )      ->     Dict:
          return {
              i: self.buffers[i] for i in range(self.num_buffers ) if i not in self.empty
          }






       def     UpperCamelCase			(							self )      ->     Tuple:
          for i in range(self.num_buffers ):
             if self.buffers[i] is None and i not in self.empty:
                _SCREAMING_SNAKE_CASE       	=			self.files[i].readline()

                if self.buffers[i] == "":
                   self.empty.add(A__ )
                   self.files[i].close()

          if len(self.empty ) == self.num_buffers:
             return False

          return True






       def     UpperCamelCase			(							self    ,							A__ )      ->     List[str]:
          _SCREAMING_SNAKE_CASE       	=			self.buffers[index]
          _SCREAMING_SNAKE_CASE       	=			None

          return value






class 				_a			:





       """simple docstring"""




       def __init__(							self    ,							A__ )      ->     str:
          _SCREAMING_SNAKE_CASE       	=			merge_strategy






       def     UpperCamelCase			(							self    ,							A__    ,							A__    ,							A__ )      ->     Union[str, Any]:
          _SCREAMING_SNAKE_CASE       	=			FilesArray(self.get_file_handles(A__    ,							A__ ) )
          with open(A__    ,							"""w"""    ,							A__ ) as outfile:
             while buffers.refresh():
                _SCREAMING_SNAKE_CASE       	=			self.merge_strategy.select(buffers.get_dict() )
                outfile.write(buffers.unshift(A__ ) )






       def     UpperCamelCase			(							self    ,							A__    ,							A__ )      ->     str:
          _SCREAMING_SNAKE_CASE       	=			{}

          for i in range(len(A__ ) ):
             _SCREAMING_SNAKE_CASE       	=			open(filenames[i]    ,							"""r"""    ,							A__ )  # noqa: UP015

          return files






class 				_a			:





       """simple docstring"""




       def __init__(							self    ,							A__ )      ->     Union[str, Any]:
          _SCREAMING_SNAKE_CASE       	=			block_size






       def     UpperCamelCase			(							self    ,							A__    ,							A__=None )      ->     Optional[Any]:
          _SCREAMING_SNAKE_CASE       	=			self.get_number_blocks(A__    ,							self.block_size )
          _SCREAMING_SNAKE_CASE       	=			FileSplitter(A__ )
          splitter.split(self.block_size    ,							A__ )

          _SCREAMING_SNAKE_CASE       	=			FileMerger(NWayMerge() )
          _SCREAMING_SNAKE_CASE       	=			self.block_size / (num_blocks + 1)
          merger.merge(splitter.get_block_filenames()    ,							filename + """.out"""    ,							A__ )

          splitter.cleanup()






       def     UpperCamelCase			(							self    ,							A__    ,							A__ )      ->     Any:
          return (os.stat(A__ ).st_size / block_size) + 1






def 				lowerCAmelCase_		(			SCREAMING_SNAKE_CASE_  )       ->       str:




   """simple docstring"""
   if string[-1].lower() == "k":
      return int(string[:-1]  ) * 10_24
   elif string[-1].lower() == "m":
      return int(string[:-1]  ) * 10_24 * 10_24
   elif string[-1].lower() == "g":
      return int(string[:-1]  ) * 10_24 * 10_24 * 10_24
   else:
      return int(SCREAMING_SNAKE_CASE_  )
def 				lowerCAmelCase_		(			)       ->       List[str]:




   """simple docstring"""
   _SCREAMING_SNAKE_CASE       	=			argparse.ArgumentParser()
   parser.add_argument(
       """-m"""      , """--mem"""      , help="""amount of memory to use for sorting"""      , default="""100M"""  )
   parser.add_argument(
       """filename"""      , metavar="""<filename>"""      , nargs=1      , help="""name of file to sort"""  )
   _SCREAMING_SNAKE_CASE       	=			parser.parse_args()

   _SCREAMING_SNAKE_CASE       	=			ExternalSort(parse_memory(args.mem  )  )
   sorter.sort(args.filename[0]  )


if __name__ == "__main__":
     main()






