


import unittest

from diffusers.models.unet_ad_blocks import *  # noqa F403
from diffusers.utils import torch_device

from .test_unet_blocks_common import UNetBlockTesterMixin



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Any    = DownBlockaD  # noqa F405
   a_				:							Dict    = "down"

   def   _snake_case    (  self  : str							):






         '''simple docstring'''



         __lowerCamelCase     :      List[Any]       						=						[-0.0_232, -0.9_869, 0.8_054, -0.0_637, -0.1_688, -1.4_264, 0.4_470, -1.3_394, 0.0_904]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							int    = ResnetDownsampleBlockaD  # noqa F405
   a_				:							Optional[Any]    = "down"

   def   _snake_case    (  self  : List[str]							):






         '''simple docstring'''



         __lowerCamelCase     :      Tuple       						=						[0.0_710, 0.2_410, -0.7_320, -1.0_757, -1.1_343, 0.3_540, -0.0_133, -0.2_576, 0.0_948]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							List[str]    = AttnDownBlockaD  # noqa F405
   a_				:							Optional[int]    = "down"

   def   _snake_case    (  self  : List[str]							):






         '''simple docstring'''



         __lowerCamelCase     :      List[str]       						=						[0.0_636, 0.8_964, -0.6_234, -1.0_131, 0.0_844, 0.4_935, 0.3_437, 0.0_911, -0.2_957]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Union[str, Any]    = CrossAttnDownBlockaD  # noqa F405
   a_				:							Optional[Any]    = "down"

   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[Any]       						=						super().prepare_init_args_and_inputs_for_common()
         __lowerCamelCase     :      Optional[Any]       						=						3_2
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Any							):






         '''simple docstring'''



         __lowerCamelCase     :      Union[str, Any]       						=						[0.2_238, -0.7_396, -0.2_255, -0.3_829, 0.1_925, 1.1_665, 0.0_603, -0.7_295, 0.1_983]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							int    = SimpleCrossAttnDownBlockaD  # noqa F405
   a_				:							List[str]    = "down"

   @property
   def   _snake_case    (  self  : List[str]							):






         '''simple docstring'''



         return super().get_dummy_input(include_encoder_hidden_states=_lowerCamelCase							)

   def   _snake_case    (  self  : List[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      List[Any]       						=						super().prepare_init_args_and_inputs_for_common()
         __lowerCamelCase     :      Union[str, Any]       						=						3_2
         return init_dict, inputs_dict


   @unittest.skipIf(torch_device == """mps"""       ,  """MPS result is not consistent"""							)
   def   _snake_case    (  self  : str							):






         '''simple docstring'''



         __lowerCamelCase     :      List[Any]       						=						[0.7_921, -0.0_992, -0.1_962, -0.7_695, -0.4_242, 0.7_804, 0.4_737, 0.2_765, 0.3_338]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Any    = SkipDownBlockaD  # noqa F405
   a_				:							str    = "down"

   @property
   def   _snake_case    (  self  : List[Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_skip_sample=_lowerCamelCase							)


   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      int       						=						[-0.0_845, -0.2_087, -0.2_465, 0.0_971, 0.1_900, -0.0_484, 0.2_664, 0.4_179, 0.5_069]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Optional[int]    = AttnSkipDownBlockaD  # noqa F405
   a_				:							Optional[int]    = "down"

   @property
   def   _snake_case    (  self  : int							):






         '''simple docstring'''



         return super().get_dummy_input(include_skip_sample=_lowerCamelCase							)


   def   _snake_case    (  self  : Dict							):






         '''simple docstring'''



         __lowerCamelCase     :      List[str]       						=						[0.5_539, 0.1_609, 0.4_924, 0.0_537, -0.1_995, 0.4_050, 0.0_979, -0.2_721, -0.0_642]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							List[Any]    = DownEncoderBlockaD  # noqa F405
   a_				:							Union[str, Any]    = "down"

   @property
   def   _snake_case    (  self  : Optional[int]							):






         '''simple docstring'''



         return super().get_dummy_input(include_temb=_lowerCamelCase							)

   def   _snake_case    (  self  : Union[str, Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      str       						=						{
             """in_channels""": 3_2,
             """out_channels""": 3_2,
         }
         __lowerCamelCase     :      str       						=						self.dummy_input
         return init_dict, inputs_dict


   def   _snake_case    (  self  : str							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[int]       						=						[1.1_102, 0.5_302, 0.4_872, -0.0_023, -0.8_042, 0.0_483, -0.3_489, -0.5_632, 0.7_626]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							int    = AttnDownEncoderBlockaD  # noqa F405
   a_				:							Union[str, Any]    = "down"

   @property
   def   _snake_case    (  self  : str							):






         '''simple docstring'''



         return super().get_dummy_input(include_temb=_lowerCamelCase							)

   def   _snake_case    (  self  : int							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[int]       						=						{
             """in_channels""": 3_2,
             """out_channels""": 3_2,
         }
         __lowerCamelCase     :      Optional[Any]       						=						self.dummy_input
         return init_dict, inputs_dict


   def   _snake_case    (  self  : List[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Dict       						=						[0.8_966, -0.1_486, 0.8_568, 0.8_141, -0.9_046, -0.1_342, -0.0_972, -0.7_417, 0.1_538]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							str    = UNetMidBlockaD  # noqa F405
   a_				:							Any    = "mid"

   def   _snake_case    (  self  : Any							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[Any]       						=						{
             """in_channels""": 3_2,
             """temb_channels""": 1_2_8,
         }
         __lowerCamelCase     :      Any       						=						self.dummy_input
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Tuple							):






         '''simple docstring'''



         __lowerCamelCase     :      int       						=						[-0.1_062, 1.7_248, 0.3_494, 1.4_569, -0.0_910, -1.2_421, -0.9_984, 0.6_736, 1.0_028]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							List[Any]    = UNetMidBlockaDCrossAttn  # noqa F405
   a_				:							str    = "mid"

   def   _snake_case    (  self  : Any							):






         '''simple docstring'''



         __lowerCamelCase     :      Dict       						=						super().prepare_init_args_and_inputs_for_common()
         __lowerCamelCase     :      Any       						=						3_2
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Tuple							):






         '''simple docstring'''



         __lowerCamelCase     :      Tuple       						=						[0.0_187, 2.4_220, 0.4_484, 1.1_203, -0.6_121, -1.5_122, -0.8_270, 0.7_851, 1.8_335]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Any    = UNetMidBlockaDSimpleCrossAttn  # noqa F405
   a_				:							Optional[Any]    = "mid"

   @property
   def   _snake_case    (  self  : Tuple							):






         '''simple docstring'''



         return super().get_dummy_input(include_encoder_hidden_states=_lowerCamelCase							)

   def   _snake_case    (  self  : Any							):






         '''simple docstring'''



         __lowerCamelCase     :      List[Any]       						=						super().prepare_init_args_and_inputs_for_common()
         __lowerCamelCase     :      str       						=						3_2
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Any       						=						[0.7_143, 1.9_974, 0.5_448, 1.3_977, 0.1_282, -1.1_237, -1.4_238, 0.5_530, 0.8_880]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							List[Any]    = UpBlockaD  # noqa F405
   a_				:							List[Any]    = "up"

   @property
   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_res_hidden_states_tuple=_lowerCamelCase							)


   def   _snake_case    (  self  : Optional[int]							):






         '''simple docstring'''



         __lowerCamelCase     :      Any       						=						[-0.2_041, -0.4_165, -0.3_022, 0.0_041, -0.6_628, -0.7_053, 0.1_928, -0.0_325, 0.0_523]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							List[Any]    = ResnetUpsampleBlockaD  # noqa F405
   a_				:							Union[str, Any]    = "up"

   @property
   def   _snake_case    (  self  : Union[str, Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_res_hidden_states_tuple=_lowerCamelCase							)


   def   _snake_case    (  self  : Optional[int]							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[int]       						=						[0.2_287, 0.3_549, -0.1_346, 0.4_797, -0.1_715, -0.9_649, 0.7_305, -0.5_864, -0.6_244]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Optional[int]    = CrossAttnUpBlockaD  # noqa F405
   a_				:							List[Any]    = "up"

   @property
   def   _snake_case    (  self  : Dict							):






         '''simple docstring'''



         return super().get_dummy_input(include_res_hidden_states_tuple=_lowerCamelCase							)

   def   _snake_case    (  self  : str							):






         '''simple docstring'''



         __lowerCamelCase     :      Any       						=						super().prepare_init_args_and_inputs_for_common()
         __lowerCamelCase     :      Tuple       						=						3_2
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Optional[int]							):






         '''simple docstring'''



         __lowerCamelCase     :      Tuple       						=						[-0.1_403, -0.3_515, -0.0_420, -0.1_425, 0.3_167, 0.5_094, -0.2_181, 0.5_931, 0.5_582]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							int    = SimpleCrossAttnUpBlockaD  # noqa F405
   a_				:							Optional[int]    = "up"

   @property
   def   _snake_case    (  self  : Optional[int]							):






         '''simple docstring'''



         return super().get_dummy_input(include_res_hidden_states_tuple=_lowerCamelCase       ,  include_encoder_hidden_states=_lowerCamelCase							)

   def   _snake_case    (  self  : int							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[int]       						=						super().prepare_init_args_and_inputs_for_common()
         __lowerCamelCase     :      Tuple       						=						3_2
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Optional[int]       						=						[0.2_645, 0.1_480, 0.0_909, 0.8_044, -0.9_758, -0.9_083, 0.0_994, -1.1_453, -0.7_402]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Optional[Any]    = AttnUpBlockaD  # noqa F405
   a_				:							Optional[Any]    = "up"

   @property
   def   _snake_case    (  self  : Union[str, Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_res_hidden_states_tuple=_lowerCamelCase							)


   @unittest.skipIf(torch_device == """mps"""       ,  """MPS result is not consistent"""							)
   def   _snake_case    (  self  : Any							):






         '''simple docstring'''



         __lowerCamelCase     :      str       						=						[0.0_979, 0.1_326, 0.0_021, 0.0_659, 0.2_249, 0.0_059, 0.1_132, 0.5_952, 0.1_033]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Optional[int]    = SkipUpBlockaD  # noqa F405
   a_				:							Optional[int]    = "up"

   @property
   def   _snake_case    (  self  : List[Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_res_hidden_states_tuple=_lowerCamelCase							)


   def   _snake_case    (  self  : List[str]							):






         '''simple docstring'''



         __lowerCamelCase     :      List[str]       						=						[-0.0_893, -0.1_234, -0.1_506, -0.0_332, 0.0_123, -0.0_211, 0.0_566, 0.0_143, 0.0_362]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							Any    = AttnSkipUpBlockaD  # noqa F405
   a_				:							Optional[Any]    = "up"

   @property
   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_res_hidden_states_tuple=_lowerCamelCase							)


   def   _snake_case    (  self  : int							):






         '''simple docstring'''



         __lowerCamelCase     :      Tuple       						=						[0.0_361, 0.0_617, 0.2_787, -0.0_350, 0.0_342, 0.3_421, -0.0_843, 0.0_913, 0.3_015]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							str    = UpDecoderBlockaD  # noqa F405
   a_				:							Any    = "up"

   @property
   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_temb=_lowerCamelCase							)

   def   _snake_case    (  self  : Union[str, Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Dict       						=						{"""in_channels""": 3_2, """out_channels""": 3_2}

         __lowerCamelCase     :      int       						=						self.dummy_input
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Union[str, Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      str       						=						[0.4_404, 0.1_998, -0.9_886, -0.3_320, -0.3_128, -0.7_034, -0.6_955, -0.2_338, -0.3_137]
         super().test_output(_lowerCamelCase							)



class 							_UpperCamelCase (						A,unittest.TestCase						):

   '''simple docstring'''







   a_				:							str    = AttnUpDecoderBlockaD  # noqa F405
   a_				:							Any    = "up"

   @property
   def   _snake_case    (  self  : List[Any]							):






         '''simple docstring'''



         return super().get_dummy_input(include_temb=_lowerCamelCase							)

   def   _snake_case    (  self  : Optional[Any]							):






         '''simple docstring'''



         __lowerCamelCase     :      Dict       						=						{"""in_channels""": 3_2, """out_channels""": 3_2}

         __lowerCamelCase     :      int       						=						self.dummy_input
         return init_dict, inputs_dict


   def   _snake_case    (  self  : Tuple							):






         '''simple docstring'''



         __lowerCamelCase     :      List[str]       						=						[0.6_738, 0.4_491, 0.1_055, 1.0_710, 0.7_316, 0.3_339, 0.3_352, 0.1_023, 0.3_568]
         super().test_output(_lowerCamelCase							)







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


__UpperCamelCase    :						Optional[Any]     =							{
    'configuration_swinv2': ['SWINV2_PRETRAINED_CONFIG_ARCHIVE_MAP', 'Swinv2Config'],
}


try:
    if not is_torch_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    pass
else:
    __UpperCamelCase    :						List[Any]     =							[
        'SWINV2_PRETRAINED_MODEL_ARCHIVE_LIST',
        'Swinv2ForImageClassification',
        'Swinv2ForMaskedImageModeling',
        'Swinv2Model',
        'Swinv2PreTrainedModel',
    ]


if TYPE_CHECKING:
    from .configuration_swinva import SWINV2_PRETRAINED_CONFIG_ARCHIVE_MAP, SwinvaConfig

    try:
        if not is_torch_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        pass
    else:
        from .modeling_swinva import (
            SWINV2_PRETRAINED_MODEL_ARCHIVE_LIST,
            SwinvaForImageClassification,
            SwinvaForMaskedImageModeling,
            SwinvaModel,
            SwinvaPreTrainedModel,
        )


else:
    import sys

    __UpperCamelCase    :						List[Any]     =							_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)



