




import random
import unittest

import torch

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

from ..pipeline_params import (
    TEXT_GUIDED_IMAGE_INPAINTING_BATCH_PARAMS,
    TEXT_GUIDED_IMAGE_INPAINTING_PARAMS,
)
from ..test_pipelines_common import PipelineTesterMixin
from . import IFPipelineTesterMixin


@skip_mps
class 							__UpperCamelCase			( __UpperCAmelCase     ,					__UpperCAmelCase     ,					unittest.TestCase  ):





				'''simple docstring'''



				__a							: Tuple										=IFInpaintingSuperResolutionPipeline
				__a							: Dict										=TEXT_GUIDED_IMAGE_INPAINTING_PARAMS - {"""width""", """height"""}
				__a							: int										=TEXT_GUIDED_IMAGE_INPAINTING_BATCH_PARAMS.union({"""original_image"""}  )
				__a							: Union[str, Any]										=PipelineTesterMixin.required_optional_params - {"""latents"""}



				def       __snake_case		( self						):
											return self._get_superresolution_dummy_components()



				def       __snake_case		( self       , UpperCAmelCase_       , UpperCAmelCase_=0						):
											if str(UpperCAmelCase_						).startswith('''mps'''						):
																		lowerCAmelCase   =		torch.manual_seed(UpperCAmelCase_						)
											else:
																		lowerCAmelCase   =		torch.Generator(device=UpperCAmelCase_						).manual_seed(UpperCAmelCase_						)

											lowerCAmelCase   =		floats_tensor((1, 3, 16, 16)       , rng=random.Random(UpperCAmelCase_						)						).to(UpperCAmelCase_						)
											lowerCAmelCase   =		floats_tensor((1, 3, 32, 32)       , rng=random.Random(UpperCAmelCase_						)						).to(UpperCAmelCase_						)
											lowerCAmelCase   =		floats_tensor((1, 3, 32, 32)       , rng=random.Random(UpperCAmelCase_						)						).to(UpperCAmelCase_						)

											lowerCAmelCase   =		{
											    '''prompt''': '''A painting of a squirrel eating a burger''',
											    '''image''': image,
											    '''original_image''': original_image,
											    '''mask_image''': mask_image,
											    '''generator''': generator,
											    '''num_inference_steps''': 2,
											    '''output_type''': '''numpy''',
											}

											return inputs



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



				def       __snake_case		( self						):
											self._test_save_load_optional_components()



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



				def       __snake_case		( self						):
											self._test_attention_slicing_forward_pass(expected_max_diff=1E-2						)



				def       __snake_case		( self						):
											self._test_save_load_local()






				def       __snake_case		( self						):
											self._test_inference_batch_single_identical(
											    expected_max_diff=1E-2       , )









from ...configuration_utils import PretrainedConfig
from ...utils import logging
from ...utils.backbone_utils import BackboneConfigMixin, get_aligned_output_features_output_indices


UpperCAmelCase_						=logging.get_logger(__name__)


class 							__UpperCamelCase			( __UpperCAmelCase     ,					__UpperCAmelCase  ):





				'''simple docstring'''



				__a							: Optional[Any]										="""maskformer-swin"""

				__a							: Optional[int]										={
				    """num_attention_heads""": """num_heads""",
				    """num_hidden_layers""": """num_layers""",
				}



				def __init__( self       , UpperCAmelCase_=2_24       , UpperCAmelCase_=4       , UpperCAmelCase_=3       , UpperCAmelCase_=96       , UpperCAmelCase_=[2, 2, 6, 2]       , UpperCAmelCase_=[3, 6, 12, 24]       , UpperCAmelCase_=7       , UpperCAmelCase_=4.0       , UpperCAmelCase_=True       , UpperCAmelCase_=0.0       , UpperCAmelCase_=0.0       , UpperCAmelCase_=0.1       , UpperCAmelCase_="gelu"       , UpperCAmelCase_=False       , UpperCAmelCase_=0.02       , UpperCAmelCase_=1E-5       , UpperCAmelCase_=None       , UpperCAmelCase_=None       , **UpperCAmelCase_       , ):
											super().__init__(**UpperCAmelCase_						)

											lowerCAmelCase   =		image_size
											lowerCAmelCase   =		patch_size
											lowerCAmelCase   =		num_channels
											lowerCAmelCase   =		embed_dim
											lowerCAmelCase   =		depths
											lowerCAmelCase   =		len(UpperCAmelCase_						)
											lowerCAmelCase   =		num_heads
											lowerCAmelCase   =		window_size
											lowerCAmelCase   =		mlp_ratio
											lowerCAmelCase   =		qkv_bias
											lowerCAmelCase   =		hidden_dropout_prob
											lowerCAmelCase   =		attention_probs_dropout_prob
											lowerCAmelCase   =		drop_path_rate
											lowerCAmelCase   =		hidden_act
											lowerCAmelCase   =		use_absolute_embeddings
											lowerCAmelCase   =		layer_norm_eps
											lowerCAmelCase   =		initializer_range
											# we set the hidden_size attribute in order to make Swin work with VisionEncoderDecoderModel
											# this indicates the channel dimension after the last stage of the model
											lowerCAmelCase   =		int(embed_dim * 2 ** (len(UpperCAmelCase_						) - 1)						)
											lowerCAmelCase   =		['''stem'''] + [F"""stage{idx}""" for idx in range(1       , len(UpperCAmelCase_						) + 1						)]
											lowerCAmelCase   ,     lowerCAmelCase   =		get_aligned_output_features_output_indices(
											    out_features=UpperCAmelCase_       , out_indices=UpperCAmelCase_       , stage_names=self.stage_names						)



