"""simple docstring"""






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 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Tuple													=			DownBlockaD  # noqa F405
      snake_case__    :     List[Any]													=			'down'

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[-0.0_2_3_2, -0.9_8_6_9, 0.8_0_5_4, -0.0_6_3_7, -0.1_6_8_8, -1.4_2_6_4, 0.4_4_7_0, -1.3_3_9_4, 0.0_9_0_4]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Tuple													=			ResnetDownsampleBlockaD  # noqa F405
      snake_case__    :     int													=			'down'

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.0_7_1_0, 0.2_4_1_0, -0.7_3_2_0, -1.0_7_5_7, -1.1_3_4_3, 0.3_5_4_0, -0.0_1_3_3, -0.2_5_7_6, 0.0_9_4_8]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Optional[Any]													=			AttnDownBlockaD  # noqa F405
      snake_case__    :     Optional[Any]													=			'down'

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.0_6_3_6, 0.8_9_6_4, -0.6_2_3_4, -1.0_1_3_1, 0.0_8_4_4, 0.4_9_3_5, 0.3_4_3_7, 0.0_9_1_1, -0.2_9_5_7]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     int													=			CrossAttnDownBlockaD  # noqa F405
      snake_case__    :     Tuple													=			'down'

      def        A__ (		self		):







            """simple docstring"""



            lowercase							,							lowercase														=			super().prepare_init_args_and_inputs_for_common()
            lowercase														=			32
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.2_2_3_8, -0.7_3_9_6, -0.2_2_5_5, -0.3_8_2_9, 0.1_9_2_5, 1.1_6_6_5, 0.0_6_0_3, -0.7_2_9_5, 0.1_9_8_3]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     List[str]													=			SimpleCrossAttnDownBlockaD  # noqa F405
      snake_case__    :     Optional[Any]													=			'down'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_encoder_hidden_states=__lowerCAmelCase		)

      def        A__ (		self		):







            """simple docstring"""



            lowercase							,							lowercase														=			super().prepare_init_args_and_inputs_for_common()
            lowercase														=			32
            return init_dict, inputs_dict







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







            """simple docstring"""



            lowercase														=			[0.7_9_2_1, -0.0_9_9_2, -0.1_9_6_2, -0.7_6_9_5, -0.4_2_4_2, 0.7_8_0_4, 0.4_7_3_7, 0.2_7_6_5, 0.3_3_3_8]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     int													=			SkipDownBlockaD  # noqa F405
      snake_case__    :     Optional[int]													=			'down'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_skip_sample=__lowerCAmelCase		)







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[-0.0_8_4_5, -0.2_0_8_7, -0.2_4_6_5, 0.0_9_7_1, 0.1_9_0_0, -0.0_4_8_4, 0.2_6_6_4, 0.4_1_7_9, 0.5_0_6_9]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Tuple													=			AttnSkipDownBlockaD  # noqa F405
      snake_case__    :     Tuple													=			'down'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_skip_sample=__lowerCAmelCase		)







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.5_5_3_9, 0.1_6_0_9, 0.4_9_2_4, 0.0_5_3_7, -0.1_9_9_5, 0.4_0_5_0, 0.0_9_7_9, -0.2_7_2_1, -0.0_6_4_2]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     int													=			DownEncoderBlockaD  # noqa F405
      snake_case__    :     Tuple													=			'down'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_temb=__lowerCAmelCase		)

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			{
                """in_channels""": 32,
                """out_channels""": 32,
            }
            lowercase														=			self.dummy_input
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[1.1_1_0_2, 0.5_3_0_2, 0.4_8_7_2, -0.0_0_2_3, -0.8_0_4_2, 0.0_4_8_3, -0.3_4_8_9, -0.5_6_3_2, 0.7_6_2_6]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Dict													=			AttnDownEncoderBlockaD  # noqa F405
      snake_case__    :     Dict													=			'down'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_temb=__lowerCAmelCase		)

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			{
                """in_channels""": 32,
                """out_channels""": 32,
            }
            lowercase														=			self.dummy_input
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.8_9_6_6, -0.1_4_8_6, 0.8_5_6_8, 0.8_1_4_1, -0.9_0_4_6, -0.1_3_4_2, -0.0_9_7_2, -0.7_4_1_7, 0.1_5_3_8]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     List[str]													=			UNetMidBlockaD  # noqa F405
      snake_case__    :     Optional[Any]													=			'mid'

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			{
                """in_channels""": 32,
                """temb_channels""": 128,
            }
            lowercase														=			self.dummy_input
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[-0.1_0_6_2, 1.7_2_4_8, 0.3_4_9_4, 1.4_5_6_9, -0.0_9_1_0, -1.2_4_2_1, -0.9_9_8_4, 0.6_7_3_6, 1.0_0_2_8]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Dict													=			UNetMidBlockaDCrossAttn  # noqa F405
      snake_case__    :     Dict													=			'mid'

      def        A__ (		self		):







            """simple docstring"""



            lowercase							,							lowercase														=			super().prepare_init_args_and_inputs_for_common()
            lowercase														=			32
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.0_1_8_7, 2.4_2_2_0, 0.4_4_8_4, 1.1_2_0_3, -0.6_1_2_1, -1.5_1_2_2, -0.8_2_7_0, 0.7_8_5_1, 1.8_3_3_5]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Union[str, Any]													=			UNetMidBlockaDSimpleCrossAttn  # noqa F405
      snake_case__    :     Tuple													=			'mid'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_encoder_hidden_states=__lowerCAmelCase		)

      def        A__ (		self		):







            """simple docstring"""



            lowercase							,							lowercase														=			super().prepare_init_args_and_inputs_for_common()
            lowercase														=			32
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.7_1_4_3, 1.9_9_7_4, 0.5_4_4_8, 1.3_9_7_7, 0.1_2_8_2, -1.1_2_3_7, -1.4_2_3_8, 0.5_5_3_0, 0.8_8_8_0]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Any													=			UpBlockaD  # noqa F405
      snake_case__    :     int													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_res_hidden_states_tuple=__lowerCAmelCase		)







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[-0.2_0_4_1, -0.4_1_6_5, -0.3_0_2_2, 0.0_0_4_1, -0.6_6_2_8, -0.7_0_5_3, 0.1_9_2_8, -0.0_3_2_5, 0.0_5_2_3]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Optional[Any]													=			ResnetUpsampleBlockaD  # noqa F405
      snake_case__    :     Optional[int]													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_res_hidden_states_tuple=__lowerCAmelCase		)







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.2_2_8_7, 0.3_5_4_9, -0.1_3_4_6, 0.4_7_9_7, -0.1_7_1_5, -0.9_6_4_9, 0.7_3_0_5, -0.5_8_6_4, -0.6_2_4_4]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Union[str, Any]													=			CrossAttnUpBlockaD  # noqa F405
      snake_case__    :     int													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_res_hidden_states_tuple=__lowerCAmelCase		)

      def        A__ (		self		):







            """simple docstring"""



            lowercase							,							lowercase														=			super().prepare_init_args_and_inputs_for_common()
            lowercase														=			32
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[-0.1_4_0_3, -0.3_5_1_5, -0.0_4_2_0, -0.1_4_2_5, 0.3_1_6_7, 0.5_0_9_4, -0.2_1_8_1, 0.5_9_3_1, 0.5_5_8_2]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Union[str, Any]													=			SimpleCrossAttnUpBlockaD  # noqa F405
      snake_case__    :     Any													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



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

      def        A__ (		self		):







            """simple docstring"""



            lowercase							,							lowercase														=			super().prepare_init_args_and_inputs_for_common()
            lowercase														=			32
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.2_6_4_5, 0.1_4_8_0, 0.0_9_0_9, 0.8_0_4_4, -0.9_7_5_8, -0.9_0_8_3, 0.0_9_9_4, -1.1_4_5_3, -0.7_4_0_2]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Dict													=			AttnUpBlockaD  # noqa F405
      snake_case__    :     Dict													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_res_hidden_states_tuple=__lowerCAmelCase		)







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







            """simple docstring"""



            lowercase														=			[0.0_9_7_9, 0.1_3_2_6, 0.0_0_2_1, 0.0_6_5_9, 0.2_2_4_9, 0.0_0_5_9, 0.1_1_3_2, 0.5_9_5_2, 0.1_0_3_3]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     List[str]													=			SkipUpBlockaD  # noqa F405
      snake_case__    :     Optional[Any]													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_res_hidden_states_tuple=__lowerCAmelCase		)







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[-0.0_8_9_3, -0.1_2_3_4, -0.1_5_0_6, -0.0_3_3_2, 0.0_1_2_3, -0.0_2_1_1, 0.0_5_6_6, 0.0_1_4_3, 0.0_3_6_2]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     str													=			AttnSkipUpBlockaD  # noqa F405
      snake_case__    :     List[str]													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_res_hidden_states_tuple=__lowerCAmelCase		)







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.0_3_6_1, 0.0_6_1_7, 0.2_7_8_7, -0.0_3_5_0, 0.0_3_4_2, 0.3_4_2_1, -0.0_8_4_3, 0.0_9_1_3, 0.3_0_1_5]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     Optional[int]													=			UpDecoderBlockaD  # noqa F405
      snake_case__    :     Any													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_temb=__lowerCAmelCase		)

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			{"""in_channels""": 32, """out_channels""": 32}

            lowercase														=			self.dummy_input
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.4_4_0_4, 0.1_9_9_8, -0.9_8_8_6, -0.3_3_2_0, -0.3_1_2_8, -0.7_0_3_4, -0.6_9_5_5, -0.2_3_3_8, -0.3_1_3_7]
            super().test_output(__lowerCAmelCase		)






class 				_A		( lowerCAmelCase						,					unittest.TestCase			):
      snake_case__    :     List[str]													=			AttnUpDecoderBlockaD  # noqa F405
      snake_case__    :     List[Any]													=			'up'

      @property
      def        A__ (		self		):







            """simple docstring"""



            return super().get_dummy_input(include_temb=__lowerCAmelCase		)

      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			{"""in_channels""": 32, """out_channels""": 32}

            lowercase														=			self.dummy_input
            return init_dict, inputs_dict







      def        A__ (		self		):







            """simple docstring"""



            lowercase														=			[0.6_7_3_8, 0.4_4_9_1, 0.1_0_5_5, 1.0_7_1_0, 0.7_3_1_6, 0.3_3_3_9, 0.3_3_5_2, 0.1_0_2_3, 0.3_5_6_8]
            super().test_output(__lowerCAmelCase		)







"""simple docstring"""






from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tf_available, is_torch_available


__lowerCAmelCase					: Dict        ={
    """configuration_xlm""": ["""XLM_PRETRAINED_CONFIG_ARCHIVE_MAP""", """XLMConfig""", """XLMOnnxConfig"""],
    """tokenization_xlm""": ["""XLMTokenizer"""],
}

try:
       if not is_torch_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       __lowerCAmelCase					: int        =[
           """XLM_PRETRAINED_MODEL_ARCHIVE_LIST""",
           """XLMForMultipleChoice""",
           """XLMForQuestionAnswering""",
           """XLMForQuestionAnsweringSimple""",
           """XLMForSequenceClassification""",
           """XLMForTokenClassification""",
           """XLMModel""",
           """XLMPreTrainedModel""",
           """XLMWithLMHeadModel""",
       ]

try:
       if not is_tf_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       __lowerCAmelCase					: Union[str, Any]        =[
           """TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST""",
           """TFXLMForMultipleChoice""",
           """TFXLMForQuestionAnsweringSimple""",
           """TFXLMForSequenceClassification""",
           """TFXLMForTokenClassification""",
           """TFXLMMainLayer""",
           """TFXLMModel""",
           """TFXLMPreTrainedModel""",
           """TFXLMWithLMHeadModel""",
       ]


if TYPE_CHECKING:
       from .configuration_xlm import XLM_PRETRAINED_CONFIG_ARCHIVE_MAP, XLMConfig, XLMOnnxConfig
       from .tokenization_xlm import XLMTokenizer

       try:
              if not is_torch_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_xlm import (
                  XLM_PRETRAINED_MODEL_ARCHIVE_LIST,
                  XLMForMultipleChoice,
                  XLMForQuestionAnswering,
                  XLMForQuestionAnsweringSimple,
                  XLMForSequenceClassification,
                  XLMForTokenClassification,
                  XLMModel,
                  XLMPreTrainedModel,
                  XLMWithLMHeadModel,
              )

       try:
              if not is_tf_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_tf_xlm import (
                  TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST,
                  TFXLMForMultipleChoice,
                  TFXLMForQuestionAnsweringSimple,
                  TFXLMForSequenceClassification,
                  TFXLMForTokenClassification,
                  TFXLMMainLayer,
                  TFXLMModel,
                  TFXLMPreTrainedModel,
                  TFXLMWithLMHeadModel,
              )

else:
       import sys

       __lowerCAmelCase					: Optional[int]        =_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)



