


'''simple docstring'''

import argparse
import re

import numpy as np
import requests
import torch
from huggingface_hub import hf_hub_download
from PIL import Image

from transformers import (
    SamConfig,
    SamImageProcessor,
    SamModel,
    SamProcessor,
    SamVisionConfig,
)


_UpperCAmelCase				: Optional[Any]							    =				{
    '''iou_prediction_head.layers.0''': '''iou_prediction_head.proj_in''',
    '''iou_prediction_head.layers.1''': '''iou_prediction_head.layers.0''',
    '''iou_prediction_head.layers.2''': '''iou_prediction_head.proj_out''',
    '''mask_decoder.output_upscaling.0''': '''mask_decoder.upscale_conv1''',
    '''mask_decoder.output_upscaling.1''': '''mask_decoder.upscale_layer_norm''',
    '''mask_decoder.output_upscaling.3''': '''mask_decoder.upscale_conv2''',
    '''mask_downscaling.0''': '''mask_embed.conv1''',
    '''mask_downscaling.1''': '''mask_embed.layer_norm1''',
    '''mask_downscaling.3''': '''mask_embed.conv2''',
    '''mask_downscaling.4''': '''mask_embed.layer_norm2''',
    '''mask_downscaling.6''': '''mask_embed.conv3''',
    '''point_embeddings''': '''point_embed''',
    '''pe_layer.positional_encoding_gaussian_matrix''': '''shared_embedding.positional_embedding''',
    '''image_encoder''': '''vision_encoder''',
    '''neck.0''': '''neck.conv1''',
    '''neck.1''': '''neck.layer_norm1''',
    '''neck.2''': '''neck.conv2''',
    '''neck.3''': '''neck.layer_norm2''',
    '''patch_embed.proj''': '''patch_embed.projection''',
    '''.norm''': '''.layer_norm''',
    '''blocks''': '''layers''',
}

def 							__magic_name__(						lowerCamelCase):
      __lowerCAmelCase							       =				{}
      state_dict.pop('''pixel_mean''',				lowerCAmelCase__)
      state_dict.pop('''pixel_std''',				lowerCAmelCase__)

      __lowerCAmelCase							       =				r'''.*.output_hypernetworks_mlps.(\d+).layers.(\d+).*'''

      for key, value in state_dict.items():
            for key_to_modify, new_key in KEYS_TO_MODIFY_MAPPING.items():
                  if key_to_modify in key:
                        __lowerCAmelCase							       =				key.replace(lowerCAmelCase__,				lowerCAmelCase__)

            if re.match(lowerCAmelCase__,				lowerCAmelCase__):
                  __lowerCAmelCase							       =				int(re.match(lowerCAmelCase__,				lowerCAmelCase__).group(2))
                  if layer_nb == 0:
                        __lowerCAmelCase							       =				key.replace('''layers.0''',				'''proj_in''')
                  elif layer_nb == 1:
                        __lowerCAmelCase							       =				key.replace('''layers.1''',				'''layers.0''')
                  elif layer_nb == 2:
                        __lowerCAmelCase							       =				key.replace('''layers.2''',				'''proj_out''')

            __lowerCAmelCase							       =				value

      __lowerCAmelCase							       =				model_state_dict[
          '''prompt_encoder.shared_embedding.positional_embedding'''
      ]

      return model_state_dict

def 							__magic_name__(						lowerCamelCase,				lowerCamelCase,				lowerCamelCase,				lowerCamelCase="ybelkada/segment-anything"):
      __lowerCAmelCase							       =				hf_hub_download(lowerCAmelCase__,				F"""checkpoints/{model_name}.pth""")

      if "sam_vit_b" in model_name:
            __lowerCAmelCase							       =				SamConfig()
      elif "sam_vit_l" in model_name:
            __lowerCAmelCase							       =				SamVisionConfig(
                hidden_size=1_0_2_4,				num_hidden_layers=2_4,				num_attention_heads=1_6,				global_attn_indexes=[5, 1_1, 1_7, 2_3],				)

            __lowerCAmelCase							       =				SamConfig(
                vision_config=lowerCAmelCase__,				)
      elif "sam_vit_h" in model_name:
            __lowerCAmelCase							       =				SamVisionConfig(
                hidden_size=1_2_8_0,				num_hidden_layers=3_2,				num_attention_heads=1_6,				global_attn_indexes=[7, 1_5, 2_3, 3_1],				)

            __lowerCAmelCase							       =				SamConfig(
                vision_config=lowerCAmelCase__,				)

      __lowerCAmelCase							       =				torch.load(lowerCAmelCase__,				map_location='''cpu''')
      __lowerCAmelCase							       =				replace_keys(lowerCAmelCase__)

      __lowerCAmelCase							       =				SamImageProcessor()

      __lowerCAmelCase							       =				SamProcessor(image_processor=lowerCAmelCase__)
      __lowerCAmelCase							       =				SamModel(lowerCAmelCase__)

      hf_model.load_state_dict(lowerCAmelCase__)
      __lowerCAmelCase							       =				hf_model.to('''cuda''')

      __lowerCAmelCase							       =				'''https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png'''
      __lowerCAmelCase							       =				Image.open(requests.get(lowerCAmelCase__,				stream=lowerCAmelCase__).raw).convert('''RGB''')

      __lowerCAmelCase							       =				[[[4_0_0, 6_5_0]]]
      __lowerCAmelCase							       =				[[1]]

      __lowerCAmelCase							       =				processor(images=np.array(lowerCAmelCase__),				return_tensors='''pt''').to('''cuda''')

      with torch.no_grad():
            __lowerCAmelCase							       =				hf_model(**lowerCAmelCase__)
      __lowerCAmelCase							       =				output.iou_scores.squeeze()

      if model_name == "sam_vit_h_4b8939":
            assert scores[-1].item() == 0.5_79_89_02_51_15_96_68

            __lowerCAmelCase							       =				processor(
                images=np.array(lowerCAmelCase__),				input_points=lowerCAmelCase__,				input_labels=lowerCAmelCase__,				return_tensors='''pt''').to('''cuda''')

            with torch.no_grad():
                  __lowerCAmelCase							       =				hf_model(**lowerCAmelCase__)
            __lowerCAmelCase							       =				output.iou_scores.squeeze()

            assert scores[-1].item() == 0.97_12_60_30_92_19_36_04

            __lowerCAmelCase							       =				((7_5, 2_7_5, 1_7_2_5, 8_5_0),)

            __lowerCAmelCase							       =				processor(images=np.array(lowerCAmelCase__),				input_boxes=lowerCAmelCase__,				return_tensors='''pt''').to('''cuda''')

            with torch.no_grad():
                  __lowerCAmelCase							       =				hf_model(**lowerCAmelCase__)
            __lowerCAmelCase							       =				output.iou_scores.squeeze()

            assert scores[-1].item() == 0.86_86_01_56_05_92_65_14

            # Test with 2 points and 1 image.
            __lowerCAmelCase							       =				[[[4_0_0, 6_5_0], [8_0_0, 6_5_0]]]
            __lowerCAmelCase							       =				[[1, 1]]

            __lowerCAmelCase							       =				processor(
                images=np.array(lowerCAmelCase__),				input_points=lowerCAmelCase__,				input_labels=lowerCAmelCase__,				return_tensors='''pt''').to('''cuda''')

            with torch.no_grad():
                  __lowerCAmelCase							       =				hf_model(**lowerCAmelCase__)
            __lowerCAmelCase							       =				output.iou_scores.squeeze()

            assert scores[-1].item() == 0.99_36_04_77_92_43_46_92


if __name__ == "__main__":
 _UpperCAmelCase				: List[str]							    =				argparse.ArgumentParser()
 _UpperCAmelCase				: List[Any]							    =				['''sam_vit_b_01ec64''', '''sam_vit_h_4b8939''', '''sam_vit_l_0b3195''']
 parser.add_argument(
     """--model_name""",
     default="""sam_vit_h_4b8939""",
     choices=choices,
     type=str,
     help="""Path to hf config.json of model to convert""",
 )
 parser.add_argument("""--pytorch_dump_folder_path""", default=None, type=str, help="""Path to the output PyTorch model.""")
 parser.add_argument(
     """--push_to_hub""",
     action="""store_true""",
     help="""Whether to push the model and processor to the hub after converting""",
 )
 parser.add_argument(
     """--model_hub_id""",
     default="""ybelkada/segment-anything""",
     choices=choices,
     type=str,
     help="""Path to hf config.json of model to convert""",
 )

 _UpperCAmelCase				: int							    =				parser.parse_args()

 convert_sam_checkpoint(args.model_name, args.pytorch_dump_folder_path, args.push_to_hub, args.model_hub_id)





'''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''']		)

