


'''simple docstring'''

import tempfile

import numpy as np
import torch
from transformers import AutoTokenizer, TaEncoderModel

from diffusers import DDPMScheduler, UNetaDConditionModel
from diffusers.models.attention_processor import AttnAddedKVProcessor
from diffusers.pipelines.deepfloyd_if import IFWatermarker
from diffusers.utils.testing_utils import torch_device

from ..test_pipelines_common import to_np


class       a__				:







    """simple docstring"""



    def 					_snake_case  (self		):
          torch.manual_seed(0		)
          __lowerCAmelCase							       =				TaEncoderModel.from_pretrained('''hf-internal-testing/tiny-random-t5'''		)

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				AutoTokenizer.from_pretrained('''hf-internal-testing/tiny-random-t5'''		)

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				UNetaDConditionModel(
              sample_size=32       ,			layers_per_block=1       ,			block_out_channels=[32, 64]       ,			down_block_types=[
                  '''ResnetDownsampleBlock2D''',
                  '''SimpleCrossAttnDownBlock2D''',
              ]       ,			mid_block_type='''UNetMidBlock2DSimpleCrossAttn'''       ,			up_block_types=['''SimpleCrossAttnUpBlock2D''', '''ResnetUpsampleBlock2D''']       ,			in_channels=3       ,			out_channels=6       ,			cross_attention_dim=32       ,			encoder_hid_dim=32       ,			attention_head_dim=8       ,			addition_embed_type='''text'''       ,			addition_embed_type_num_heads=2       ,			cross_attention_norm='''group_norm'''       ,			resnet_time_scale_shift='''scale_shift'''       ,			act_fn='''gelu'''       ,			)
          unet.set_attn_processor(AttnAddedKVProcessor()		)  # For reproducibility tests

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				DDPMScheduler(
              num_train_timesteps=10_00       ,			beta_schedule='''squaredcos_cap_v2'''       ,			beta_start=0.0_0_0_1       ,			beta_end=0.0_2       ,			thresholding=UpperCAmelCase_       ,			dynamic_thresholding_ratio=0.9_5       ,			sample_max_value=1.0       ,			prediction_type='''epsilon'''       ,			variance_type='''learned_range'''       ,			)

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				IFWatermarker()

          return {
              "text_encoder": text_encoder,
              "tokenizer": tokenizer,
              "unet": unet,
              "scheduler": scheduler,
              "watermarker": watermarker,
              "safety_checker": None,
              "feature_extractor": None,
          }




    def 					_snake_case  (self		):
          torch.manual_seed(0		)
          __lowerCAmelCase							       =				TaEncoderModel.from_pretrained('''hf-internal-testing/tiny-random-t5'''		)

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				AutoTokenizer.from_pretrained('''hf-internal-testing/tiny-random-t5'''		)

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				UNetaDConditionModel(
              sample_size=32       ,			layers_per_block=[1, 2]       ,			block_out_channels=[32, 64]       ,			down_block_types=[
                  '''ResnetDownsampleBlock2D''',
                  '''SimpleCrossAttnDownBlock2D''',
              ]       ,			mid_block_type='''UNetMidBlock2DSimpleCrossAttn'''       ,			up_block_types=['''SimpleCrossAttnUpBlock2D''', '''ResnetUpsampleBlock2D''']       ,			in_channels=6       ,			out_channels=6       ,			cross_attention_dim=32       ,			encoder_hid_dim=32       ,			attention_head_dim=8       ,			addition_embed_type='''text'''       ,			addition_embed_type_num_heads=2       ,			cross_attention_norm='''group_norm'''       ,			resnet_time_scale_shift='''scale_shift'''       ,			act_fn='''gelu'''       ,			class_embed_type='''timestep'''       ,			mid_block_scale_factor=1.4_1_4       ,			time_embedding_act_fn='''gelu'''       ,			time_embedding_dim=32       ,			)
          unet.set_attn_processor(AttnAddedKVProcessor()		)  # For reproducibility tests

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				DDPMScheduler(
              num_train_timesteps=10_00       ,			beta_schedule='''squaredcos_cap_v2'''       ,			beta_start=0.0_0_0_1       ,			beta_end=0.0_2       ,			thresholding=UpperCAmelCase_       ,			dynamic_thresholding_ratio=0.9_5       ,			sample_max_value=1.0       ,			prediction_type='''epsilon'''       ,			variance_type='''learned_range'''       ,			)

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				DDPMScheduler(
              num_train_timesteps=10_00       ,			beta_schedule='''squaredcos_cap_v2'''       ,			beta_start=0.0_0_0_1       ,			beta_end=0.0_2       ,			)

          torch.manual_seed(0		)
          __lowerCAmelCase							       =				IFWatermarker()

          return {
              "text_encoder": text_encoder,
              "tokenizer": tokenizer,
              "unet": unet,
              "scheduler": scheduler,
              "image_noising_scheduler": image_noising_scheduler,
              "watermarker": watermarker,
              "safety_checker": None,
              "feature_extractor": None,
          }




    def 					_snake_case  (self		):
          __lowerCAmelCase							       =				self.get_dummy_components()
          __lowerCAmelCase							       =				self.pipeline_class(**UpperCAmelCase_		)
          pipe.to(UpperCAmelCase_		)
          pipe.set_progress_bar_config(disable=UpperCAmelCase_		)

          __lowerCAmelCase							       =				self.get_dummy_inputs(UpperCAmelCase_		)

          __lowerCAmelCase							       =				inputs['prompt']
          __lowerCAmelCase							       =				inputs['generator']
          __lowerCAmelCase							       =				inputs['num_inference_steps']
          __lowerCAmelCase							       =				inputs['output_type']

          if "image" in inputs:
                __lowerCAmelCase							       =				inputs['image']
          else:
                __lowerCAmelCase							       =				None

          if "mask_image" in inputs:
                __lowerCAmelCase							       =				inputs['mask_image']
          else:
                __lowerCAmelCase							       =				None

          if "original_image" in inputs:
                __lowerCAmelCase							       =				inputs['original_image']
          else:
                __lowerCAmelCase							       =				None

          __lowerCAmelCase							       =				pipe.encode_prompt(UpperCAmelCase_		)

          # inputs with prompt converted to embeddings
          __lowerCAmelCase							       =				{
              'prompt_embeds': prompt_embeds,
              'negative_prompt_embeds': negative_prompt_embeds,
              'generator': generator,
              'num_inference_steps': num_inference_steps,
              'output_type': output_type,
          }

          if image is not None:
                __lowerCAmelCase							       =				image

          if mask_image is not None:
                __lowerCAmelCase							       =				mask_image

          if original_image is not None:
                __lowerCAmelCase							       =				original_image

          # set all optional components to None
          for optional_component in pipe._optional_components:
                setattr(UpperCAmelCase_       ,			UpperCAmelCase_       ,			UpperCAmelCase_		)

          __lowerCAmelCase							       =				pipe(**UpperCAmelCase_		)[0]

          with tempfile.TemporaryDirectory() as tmpdir:
                pipe.save_pretrained(UpperCAmelCase_		)
                __lowerCAmelCase							       =				self.pipeline_class.from_pretrained(UpperCAmelCase_		)
                pipe_loaded.to(UpperCAmelCase_		)
                pipe_loaded.set_progress_bar_config(disable=UpperCAmelCase_		)

          pipe_loaded.unet.set_attn_processor(AttnAddedKVProcessor()		)  # For reproducibility tests

          for optional_component in pipe._optional_components:
                self.assertTrue(
                    getattr(UpperCAmelCase_       ,			UpperCAmelCase_		) is None       ,			F"""`{optional_component}` did not stay set to None after loading."""       ,			)

          __lowerCAmelCase							       =				self.get_dummy_inputs(UpperCAmelCase_		)

          __lowerCAmelCase							       =				inputs['generator']
          __lowerCAmelCase							       =				inputs['num_inference_steps']
          __lowerCAmelCase							       =				inputs['output_type']

          # inputs with prompt converted to embeddings
          __lowerCAmelCase							       =				{
              'prompt_embeds': prompt_embeds,
              'negative_prompt_embeds': negative_prompt_embeds,
              'generator': generator,
              'num_inference_steps': num_inference_steps,
              'output_type': output_type,
          }

          if image is not None:
                __lowerCAmelCase							       =				image

          if mask_image is not None:
                __lowerCAmelCase							       =				mask_image

          if original_image is not None:
                __lowerCAmelCase							       =				original_image

          __lowerCAmelCase							       =				pipe_loaded(**UpperCAmelCase_		)[0]

          __lowerCAmelCase							       =				np.abs(to_np(UpperCAmelCase_		) - to_np(UpperCAmelCase_		)		).max()
          self.assertLess(UpperCAmelCase_       ,			1e-4		)




    def 					_snake_case  (self		):
          __lowerCAmelCase							       =				self.get_dummy_components()
          __lowerCAmelCase							       =				self.pipeline_class(**UpperCAmelCase_		)
          pipe.to(UpperCAmelCase_		)
          pipe.set_progress_bar_config(disable=UpperCAmelCase_		)

          __lowerCAmelCase							       =				self.get_dummy_inputs(UpperCAmelCase_		)
          __lowerCAmelCase							       =				pipe(**UpperCAmelCase_		)[0]

          with tempfile.TemporaryDirectory() as tmpdir:
                pipe.save_pretrained(UpperCAmelCase_		)
                __lowerCAmelCase							       =				self.pipeline_class.from_pretrained(UpperCAmelCase_		)
                pipe_loaded.to(UpperCAmelCase_		)
                pipe_loaded.set_progress_bar_config(disable=UpperCAmelCase_		)

          pipe_loaded.unet.set_attn_processor(AttnAddedKVProcessor()		)  # For reproducibility tests

          __lowerCAmelCase							       =				self.get_dummy_inputs(UpperCAmelCase_		)
          __lowerCAmelCase							       =				pipe_loaded(**UpperCAmelCase_		)[0]

          __lowerCAmelCase							       =				np.abs(to_np(UpperCAmelCase_		) - to_np(UpperCAmelCase_		)		).max()
          self.assertLess(UpperCAmelCase_       ,			1e-4		)





'''simple docstring'''

from ..utils import DummyObject, requires_backends


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[int]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Any	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		str	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Tuple	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[int]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[int]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[int]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		int	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		int	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)

def 							__magic_name__(						*lowerCamelCase,				**lowerCamelCase):
						requires_backends(lowerCamelCase,				['''torch'''])

def 							__magic_name__(						*lowerCamelCase,				**lowerCamelCase):
						requires_backends(lowerCamelCase,				['''torch'''])

def 							__magic_name__(						*lowerCamelCase,				**lowerCamelCase):
						requires_backends(lowerCamelCase,				['''torch'''])

def 							__magic_name__(						*lowerCamelCase,				**lowerCamelCase):
						requires_backends(lowerCamelCase,				['''torch'''])

def 							__magic_name__(						*lowerCamelCase,				**lowerCamelCase):
						requires_backends(lowerCamelCase,				['''torch'''])

def 							__magic_name__(						*lowerCamelCase,				**lowerCamelCase):
						requires_backends(lowerCamelCase,				['''torch'''])

def 							__magic_name__(						*lowerCamelCase,				**lowerCamelCase):
						requires_backends(lowerCamelCase,				['''torch'''])


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[str]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[str]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Union[str, Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[int]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		int	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		str	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		int	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Tuple	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		int	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[str]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[str]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		str	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Tuple	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Any	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[str]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		int	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Any	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		List[str]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Optional[int]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Union[str, Any]	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)


class       a__				( metaclass=__A      ):







				"""simple docstring"""



				__UpperCamelCase    :		Dict	=	['torch']




				def __init__(self       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(self       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)




				@classmethod
				def 					_snake_case  (cls       ,			*__lowercase       ,			**__lowercase		):
										requires_backends(cls       ,			['''torch''']		)

