





import collections
from typing import List, Optional, Union

from ...tokenization_utils_base import BatchEncoding
from ...utils import TensorType, add_end_docstrings, add_start_docstrings, logging
from ..bert.tokenization_bert import BertTokenizer


_UpperCamelCase				=       logging.get_logger(__name__)

_UpperCamelCase				=       {'''vocab_file''': '''vocab.txt''', '''tokenizer_file''': '''tokenizer.json'''}

_UpperCamelCase				=       {
    '''vocab_file''': {
        '''facebook/dpr-ctx_encoder-single-nq-base''': (
            '''https://huggingface.co/facebook/dpr-ctx_encoder-single-nq-base/resolve/main/vocab.txt'''
        ),
        '''facebook/dpr-ctx_encoder-multiset-base''': (
            '''https://huggingface.co/facebook/dpr-ctx_encoder-multiset-base/resolve/main/vocab.txt'''
        ),
    },
    '''tokenizer_file''': {
        '''facebook/dpr-ctx_encoder-single-nq-base''': (
            '''https://huggingface.co/facebook/dpr-ctx_encoder-single-nq-base/resolve/main/tokenizer.json'''
        ),
        '''facebook/dpr-ctx_encoder-multiset-base''': (
            '''https://huggingface.co/facebook/dpr-ctx_encoder-multiset-base/resolve/main/tokenizer.json'''
        ),
    },
}
_UpperCamelCase				=       {
    '''vocab_file''': {
        '''facebook/dpr-question_encoder-single-nq-base''': (
            '''https://huggingface.co/facebook/dpr-question_encoder-single-nq-base/resolve/main/vocab.txt'''
        ),
        '''facebook/dpr-question_encoder-multiset-base''': (
            '''https://huggingface.co/facebook/dpr-question_encoder-multiset-base/resolve/main/vocab.txt'''
        ),
    },
    '''tokenizer_file''': {
        '''facebook/dpr-question_encoder-single-nq-base''': (
            '''https://huggingface.co/facebook/dpr-question_encoder-single-nq-base/resolve/main/tokenizer.json'''
        ),
        '''facebook/dpr-question_encoder-multiset-base''': (
            '''https://huggingface.co/facebook/dpr-question_encoder-multiset-base/resolve/main/tokenizer.json'''
        ),
    },
}
_UpperCamelCase				=       {
    '''vocab_file''': {
        '''facebook/dpr-reader-single-nq-base''': (
            '''https://huggingface.co/facebook/dpr-reader-single-nq-base/resolve/main/vocab.txt'''
        ),
        '''facebook/dpr-reader-multiset-base''': (
            '''https://huggingface.co/facebook/dpr-reader-multiset-base/resolve/main/vocab.txt'''
        ),
    },
    '''tokenizer_file''': {
        '''facebook/dpr-reader-single-nq-base''': (
            '''https://huggingface.co/facebook/dpr-reader-single-nq-base/resolve/main/tokenizer.json'''
        ),
        '''facebook/dpr-reader-multiset-base''': (
            '''https://huggingface.co/facebook/dpr-reader-multiset-base/resolve/main/tokenizer.json'''
        ),
    },
}

_UpperCamelCase				=       {
    '''facebook/dpr-ctx_encoder-single-nq-base''': 512,
    '''facebook/dpr-ctx_encoder-multiset-base''': 512,
}
_UpperCamelCase				=       {
    '''facebook/dpr-question_encoder-single-nq-base''': 512,
    '''facebook/dpr-question_encoder-multiset-base''': 512,
}
_UpperCamelCase				=       {
    '''facebook/dpr-reader-single-nq-base''': 512,
    '''facebook/dpr-reader-multiset-base''': 512,
}


_UpperCamelCase				=       {
    '''facebook/dpr-ctx_encoder-single-nq-base''': {'''do_lower_case''': True},
    '''facebook/dpr-ctx_encoder-multiset-base''': {'''do_lower_case''': True},
}
_UpperCamelCase				=       {
    '''facebook/dpr-question_encoder-single-nq-base''': {'''do_lower_case''': True},
    '''facebook/dpr-question_encoder-multiset-base''': {'''do_lower_case''': True},
}
_UpperCamelCase				=       {
    '''facebook/dpr-reader-single-nq-base''': {'''do_lower_case''': True},
    '''facebook/dpr-reader-multiset-base''': {'''do_lower_case''': True},
}




class  _lowerCamelCase       (			UpperCAmelCase_			):



      """simple docstring"""






      UpperCAmelCase_  :					List[Any]      			=VOCAB_FILES_NAMES
      UpperCAmelCase_  :					Optional[int]      			=CONTEXT_ENCODER_PRETRAINED_VOCAB_FILES_MAP
      UpperCAmelCase_  :					Tuple      			=CONTEXT_ENCODER_PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
      UpperCAmelCase_  :					List[str]      			=CONTEXT_ENCODER_PRETRAINED_INIT_CONFIGURATION




class  _lowerCamelCase       (			UpperCAmelCase_			):



      """simple docstring"""






      UpperCAmelCase_  :					Dict      			=VOCAB_FILES_NAMES
      UpperCAmelCase_  :					Optional[int]      			=QUESTION_ENCODER_PRETRAINED_VOCAB_FILES_MAP
      UpperCAmelCase_  :					Optional[int]      			=QUESTION_ENCODER_PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
      UpperCAmelCase_  :					Optional[Any]      			=QUESTION_ENCODER_PRETRAINED_INIT_CONFIGURATION




_UpperCamelCase				=       collections.namedtuple(
    '''DPRSpanPrediction''', ['''span_score''', '''relevance_score''', '''doc_id''', '''start_index''', '''end_index''', '''text''']
)

_UpperCamelCase				=       collections.namedtuple('''DPRReaderOutput''', ['''start_logits''', '''end_logits''', '''relevance_logits'''])


_UpperCamelCase				=       R'''
    Return a dictionary with the token ids of the input strings and other information to give to `.decode_best_spans`.
    It converts the strings of a question and different passages (title and text) in a sequence of IDs (integers),
    using the tokenizer and vocabulary. The resulting `input_ids` is a matrix of size `(n_passages, sequence_length)`
    with the format:

    ```
    [CLS] <question token ids> [SEP] <titles ids> [SEP] <texts ids>
    ```

    Args:
        questions (`str` or `List[str]`):
            The questions to be encoded. You can specify one question for many passages. In this case, the question
            will be duplicated like `[questions] * n_passages`. Otherwise you have to specify as many questions as in
            `titles` or `texts`.
        titles (`str` or `List[str]`):
            The passages titles to be encoded. This can be a string or a list of strings if there are several passages.
        texts (`str` or `List[str]`):
            The passages texts to be encoded. This can be a string or a list of strings if there are several passages.
        padding (`bool`, `str` or [`~utils.PaddingStrategy`], *optional*, defaults to `False`):
            Activates and controls padding. Accepts the following values:

            - `True` or `\'longest\'`: Pad to the longest sequence in the batch (or no padding if only a single sequence
              if provided).
            - `\'max_length\'`: Pad to a maximum length specified with the argument `max_length` or to the maximum
              acceptable input length for the model if that argument is not provided.
            - `False` or `\'do_not_pad\'` (default): No padding (i.e., can output a batch with sequences of different
              lengths).
        truncation (`bool`, `str` or [`~tokenization_utils_base.TruncationStrategy`], *optional*, defaults to `False`):
            Activates and controls truncation. Accepts the following values:

            - `True` or `\'longest_first\'`: Truncate to a maximum length specified with the argument `max_length` or to
              the maximum acceptable input length for the model if that argument is not provided. This will truncate
              token by token, removing a token from the longest sequence in the pair if a pair of sequences (or a batch
              of pairs) is provided.
            - `\'only_first\'`: Truncate to a maximum length specified with the argument `max_length` or to the maximum
              acceptable input length for the model if that argument is not provided. This will only truncate the first
              sequence of a pair if a pair of sequences (or a batch of pairs) is provided.
            - `\'only_second\'`: Truncate to a maximum length specified with the argument `max_length` or to the maximum
              acceptable input length for the model if that argument is not provided. This will only truncate the
              second sequence of a pair if a pair of sequences (or a batch of pairs) is provided.
            - `False` or `\'do_not_truncate\'` (default): No truncation (i.e., can output batch with sequence lengths
              greater than the model maximum admissible input size).
        max_length (`int`, *optional*):
                Controls the maximum length to use by one of the truncation/padding parameters.

                If left unset or set to `None`, this will use the predefined model maximum length if a maximum length
                is required by one of the truncation/padding parameters. If the model has no specific maximum input
                length (like XLNet) truncation/padding to a maximum length will be deactivated.
        return_tensors (`str` or [`~utils.TensorType`], *optional*):
                If set, will return tensors instead of list of python integers. Acceptable values are:

                - `\'tf\'`: Return TensorFlow `tf.constant` objects.
                - `\'pt\'`: Return PyTorch `torch.Tensor` objects.
                - `\'np\'`: Return Numpy `np.ndarray` objects.
        return_attention_mask (`bool`, *optional*):
            Whether or not to return the attention mask. If not set, will return the attention mask according to the
            specific tokenizer\'s default, defined by the `return_outputs` attribute.

            [What are attention masks?](../glossary#attention-mask)

    Returns:
        `Dict[str, List[List[int]]]`: A dictionary with the following keys:

        - `input_ids`: List of token ids to be fed to a model.
        - `attention_mask`: List of indices specifying which tokens should be attended to by the model.
    '''




@add_start_docstrings(UpperCAmelCase_			)
class  _lowerCamelCase       :



      """simple docstring"""






      def __call__(       self ,  UpperCAmelCase ,  UpperCAmelCase = None ,  UpperCAmelCase = None ,  UpperCAmelCase = False ,  UpperCAmelCase = False ,  UpperCAmelCase = None ,  UpperCAmelCase = None ,  UpperCAmelCase = None ,  **UpperCAmelCase ,  )				->      BatchEncoding:






        '''simple docstring'''






        if titles is None and texts is None:
          return super().__call__(
              __lowercase ,  padding=__lowercase ,  truncation=__lowercase ,  max_length=__lowercase ,  return_tensors=__lowercase ,  return_attention_mask=__lowercase ,  **__lowercase ,  )
        elif titles is None or texts is None:
          __snake_case      : List[Any]						=				titles if texts is None else texts
          return super().__call__(
              __lowercase ,  __lowercase ,  padding=__lowercase ,  truncation=__lowercase ,  max_length=__lowercase ,  return_tensors=__lowercase ,  return_attention_mask=__lowercase ,  **__lowercase ,  )
        __snake_case      : List[Any]						=				titles if not isinstance(__lowercase ,  __lowercase      ) else [titles]
        __snake_case      : int						=				texts if not isinstance(__lowercase ,  __lowercase      ) else [texts]
        __snake_case      : Optional[Any]						=				len(__lowercase      )
        __snake_case      : Union[str, Any]						=				questions if not isinstance(__lowercase ,  __lowercase      ) else [questions] * n_passages
        if len(__lowercase      ) != len(__lowercase      ):
          raise ValueError(
              F"""There should be as many titles than texts but got {len(__lowercase      )} titles and {len(__lowercase      )} texts."""      )
        __snake_case      : Optional[int]						=				super().__call__(__lowercase ,  __lowercase ,  padding=__lowercase ,  truncation=__lowercase      )['''input_ids''']
        __snake_case      : Tuple						=				super().__call__(__lowercase ,  add_special_tokens=__lowercase ,  padding=__lowercase ,  truncation=__lowercase      )['''input_ids''']
        __snake_case      : Optional[Any]						=				{
            '''input_ids''': [
                (encoded_question_and_title + encoded_text)[:max_length]
                if max_length is not None and truncation
                else encoded_question_and_title + encoded_text
                for encoded_question_and_title, encoded_text in zip(__lowercase ,  __lowercase      )
            ]
        }
        if return_attention_mask is not False:
          __snake_case      : List[str]						=				[]
          for input_ids in encoded_inputs["input_ids"]:
            attention_mask.append([int(input_id != self.pad_token_id      ) for input_id in input_ids]      )
          __snake_case      : Union[str, Any]						=				attention_mask
        return self.pad(__lowercase ,  padding=__lowercase ,  max_length=__lowercase ,  return_tensors=__lowercase      )







      def   UpperCAmelCase						(       self ,  UpperCAmelCase ,  UpperCAmelCase ,  UpperCAmelCase = 16 ,  UpperCAmelCase = 64 ,  UpperCAmelCase = 4 ,  )				->      List[DPRSpanPrediction]:






        '''simple docstring'''






        __snake_case      : Union[str, Any]						=				reader_input['''input_ids''']
        __snake_case      : Any						=				reader_output[:3]
        __snake_case      : Optional[Any]						=				len(__lowercase      )
        __snake_case      : List[Any]						=				sorted(range(__lowercase      ) ,  reverse=__lowercase ,  key=relevance_logits.__getitem__      )
        __snake_case      : List[DPRReaderOutput]						=				[]
        for doc_id in sorted_docs:
          __snake_case      : Dict						=				list(input_ids[doc_id]      )
          # assuming question & title information is at the beginning of the sequence
          __snake_case      : List[Any]						=				sequence_ids.index(self.sep_token_id ,  2      ) + 1  # second sep id
          if sequence_ids[-1] == self.pad_token_id:
            __snake_case      : Any						=				sequence_ids.index(self.pad_token_id      )
          else:
            __snake_case      : Optional[int]						=				len(__lowercase      )

          __snake_case      : Dict						=				self._get_best_spans(
              start_logits=start_logits[doc_id][passage_offset:sequence_len] ,  end_logits=end_logits[doc_id][passage_offset:sequence_len] ,  max_answer_length=__lowercase ,  top_spans=__lowercase ,  )
          for start_index, end_index in best_spans:
            start_index += passage_offset
            end_index += passage_offset
            nbest_spans_predictions.append(
                DPRSpanPrediction(
                    span_score=start_logits[doc_id][start_index] + end_logits[doc_id][end_index] ,  relevance_score=relevance_logits[doc_id] ,  doc_id=__lowercase ,  start_index=__lowercase ,  end_index=__lowercase ,  text=self.decode(sequence_ids[start_index : end_index + 1]      ) ,  )      )
          if len(__lowercase      ) >= num_spans:
            break
        return nbest_spans_predictions[:num_spans]







      def   UpperCAmelCase						(       self ,  UpperCAmelCase ,  UpperCAmelCase ,  UpperCAmelCase ,  UpperCAmelCase ,  )				->      List[DPRSpanPrediction]:






        '''simple docstring'''






        __snake_case      : Optional[int]						=				[]
        for start_index, start_score in enumerate(__lowercase      ):
          for answer_length, end_score in enumerate(end_logits[start_index : start_index + max_answer_length]      ):
            scores.append(((start_index, start_index + answer_length), start_score + end_score)      )
        __snake_case      : Tuple						=				sorted(__lowercase ,  key=lambda UpperCAmelCase      : x[1] ,  reverse=__lowercase      )
        __snake_case      : str						=				[]
        for (start_index, end_index), score in scores:
          if start_index > end_index:
            raise ValueError(F"""Wrong span indices: [{start_index}:{end_index}]"""      )
          __snake_case      : List[str]						=				end_index - start_index + 1
          if length > max_answer_length:
            raise ValueError(F"""Span is too long: {length} > {max_answer_length}"""      )
          if any(
              start_index <= prev_start_index <= prev_end_index <= end_index
              or prev_start_index <= start_index <= end_index <= prev_end_index
              for (prev_start_index, prev_end_index) in chosen_span_intervals      ):
            continue
          chosen_span_intervals.append((start_index, end_index)      )

          if len(__lowercase      ) == top_spans:
            break
        return chosen_span_intervals




@add_end_docstrings(UpperCAmelCase_			)
class  _lowerCamelCase       (			UpperCAmelCase_       ,      UpperCAmelCase_			):



      """simple docstring"""






      UpperCAmelCase_  :					Optional[int]      			=VOCAB_FILES_NAMES
      UpperCAmelCase_  :					Dict      			=READER_PRETRAINED_VOCAB_FILES_MAP
      UpperCAmelCase_  :					Any      			=READER_PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
      UpperCAmelCase_  :					int      			=READER_PRETRAINED_INIT_CONFIGURATION
      UpperCAmelCase_  :					int      			=["""input_ids""", """attention_mask"""]


import gc
import random
import unittest

import numpy as np
import torch
from transformers import (
    CLIPImageProcessor,
    CLIPTextConfig,
    CLIPTextModel,
    CLIPTokenizer,
    CLIPVisionConfig,
    CLIPVisionModelWithProjection,
)

from diffusers import AutoencoderKL, DDIMScheduler, DDPMScheduler, StableUnCLIPImgaImgPipeline, UNetaDConditionModel
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
from diffusers.pipelines.stable_diffusion.stable_unclip_image_normalizer import StableUnCLIPImageNormalizer
from diffusers.utils.import_utils import is_xformers_available
from diffusers.utils.testing_utils import (
    enable_full_determinism,
    floats_tensor,
    load_image,
    load_numpy,
    require_torch_gpu,
    skip_mps,
    slow,
    torch_device,
)

from ..pipeline_params import TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS, TEXT_GUIDED_IMAGE_VARIATION_PARAMS
from ..test_pipelines_common import (
    PipelineKarrasSchedulerTesterMixin,
    PipelineLatentTesterMixin,
    PipelineTesterMixin,
    assert_mean_pixel_difference,
)


enable_full_determinism()


class 	lowerCamelCase_     (   UpperCAmelCase_						,	UpperCAmelCase_						,	UpperCAmelCase_						,	unittest.TestCase ):






    '''simple docstring'''






    a__	:						int						=      StableUnCLIPImgaImgPipeline
    a__	:						Optional[int]						=      TEXT_GUIDED_IMAGE_VARIATION_PARAMS
    a__	:						Union[str, Any]						=      TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS
    a__	:						Optional[Any]						=      frozenset(
        [] )  # TO-DO: update image_params once pipeline is refactored with VaeImageProcessor.preprocess
    a__	:						int						=      frozenset([] )







    def 			UpperCamelCase__   (	self)       ->	Tuple:
        __UpperCamelCase     :Tuple			  =		32
        __UpperCamelCase     :Optional[int]			  =		embedder_hidden_size

        # image encoding components

        __UpperCamelCase     :Union[str, Any]			  =		CLIPImageProcessor(crop_size=32 , size=32)

        torch.manual_seed(0)
        __UpperCamelCase     :Union[str, Any]			  =		CLIPVisionModelWithProjection(
            CLIPVisionConfig(
                hidden_size=__lowercase , projection_dim=__lowercase , num_hidden_layers=5 , num_attention_heads=4 , image_size=32 , intermediate_size=37 , patch_size=1 , ))

        # regular denoising components

        torch.manual_seed(0)
        __UpperCamelCase     :str			  =		StableUnCLIPImageNormalizer(embedding_dim=__lowercase)
        __UpperCamelCase     :Optional[int]			  =		DDPMScheduler(beta_schedule='''squaredcos_cap_v2''')

        torch.manual_seed(0)
        __UpperCamelCase     :Union[str, Any]			  =		CLIPTokenizer.from_pretrained('''hf-internal-testing/tiny-random-clip''')

        torch.manual_seed(0)
        __UpperCamelCase     :Dict			  =		CLIPTextModel(
            CLIPTextConfig(
                bos_token_id=0 , eos_token_id=2 , hidden_size=__lowercase , projection_dim=32 , intermediate_size=37 , layer_norm_eps=1E-0_5 , num_attention_heads=4 , num_hidden_layers=5 , pad_token_id=1 , vocab_size=1_000 , ))

        torch.manual_seed(0)
        __UpperCamelCase     :List[Any]			  =		UNetaDConditionModel(
            sample_size=32 , in_channels=4 , out_channels=4 , down_block_types=('''CrossAttnDownBlock2D''', '''DownBlock2D''') , up_block_types=('''UpBlock2D''', '''CrossAttnUpBlock2D''') , block_out_channels=(32, 64) , attention_head_dim=(2, 4) , class_embed_type='''projection''' , projection_class_embeddings_input_dim=embedder_projection_dim * 2 , cross_attention_dim=__lowercase , layers_per_block=1 , upcast_attention=__lowercase , use_linear_projection=__lowercase , )

        torch.manual_seed(0)
        __UpperCamelCase     :Tuple			  =		DDIMScheduler(
            beta_schedule='''scaled_linear''' , beta_start=0.0_00_85 , beta_end=0.0_12 , prediction_type='''v_prediction''' , set_alpha_to_one=__lowercase , steps_offset=1 , )

        torch.manual_seed(0)
        __UpperCamelCase     :List[str]			  =		AutoencoderKL()

        __UpperCamelCase     :Tuple			  =		{
            # image encoding components
            '''feature_extractor''': feature_extractor,
            '''image_encoder''': image_encoder.eval(),
            # image noising components
            '''image_normalizer''': image_normalizer.eval(),
            '''image_noising_scheduler''': image_noising_scheduler,
            # regular denoising components
            '''tokenizer''': tokenizer,
            '''text_encoder''': text_encoder.eval(),
            '''unet''': unet.eval(),
            '''scheduler''': scheduler,
            '''vae''': vae.eval(),
        }

        return components







    def 			UpperCamelCase__   (	self , __lowercase , __lowercase=0 , __lowercase=True)       ->	str:
        if str(__lowercase).startswith('''mps'''):
            __UpperCamelCase     :Union[str, Any]			  =		torch.manual_seed(__lowercase)
        else:
            __UpperCamelCase     :int			  =		torch.Generator(device=__lowercase).manual_seed(__lowercase)

        __UpperCamelCase     :int			  =		floats_tensor((1, 3, 32, 32) , rng=random.Random(__lowercase)).to(__lowercase)

        if pil_image:
            __UpperCamelCase     :List[Any]			  =		input_image * 0.5 + 0.5
            __UpperCamelCase     :Optional[Any]			  =		input_image.clamp(0 , 1)
            __UpperCamelCase     :int			  =		input_image.cpu().permute(0 , 2 , 3 , 1).float().numpy()
            __UpperCamelCase     :Optional[Any]			  =		DiffusionPipeline.numpy_to_pil(__lowercase)[0]

        return {
            "prompt": "An anime racoon running a marathon",
            "image": input_image,
            "generator": generator,
            "num_inference_steps": 2,
            "output_type": "np",
        }







    @skip_mps
    def 			UpperCamelCase__   (	self)       ->	Union[str, Any]:
        __UpperCamelCase     :Dict			  =		'''cpu'''  # ensure determinism for the device-dependent torch.Generator
        __UpperCamelCase     :Tuple			  =		self.get_dummy_components()
        __UpperCamelCase     :Any			  =		StableUnCLIPImgaImgPipeline(**__lowercase)
        __UpperCamelCase     :Optional[Any]			  =		sd_pipe.to(__lowercase)
        sd_pipe.set_progress_bar_config(disable=__lowercase)

        __UpperCamelCase     :List[Any]			  =		self.get_dummy_inputs(__lowercase)
        inputs.update({'''image_embeds''': None})
        __UpperCamelCase     :Any			  =		sd_pipe(**__lowercase).images
        __UpperCamelCase     :List[str]			  =		image[0, -3:, -3:, -1]

        assert image.shape == (1, 32, 32, 3)
        __UpperCamelCase     :List[Any]			  =		np.array([0.38_72, 0.72_24, 0.56_01, 0.47_41, 0.68_72, 0.58_14, 0.46_36, 0.38_67, 0.50_78])

        assert np.abs(image_slice.flatten() - expected_slice).max() < 1E-3







    def 			UpperCamelCase__   (	self)       ->	str:
        __UpperCamelCase     :Optional[Any]			  =		torch_device in ['''cpu''', '''mps''']

        self._test_attention_slicing_forward_pass(test_max_difference=__lowercase)







    def 			UpperCamelCase__   (	self)       ->	List[Any]:
        __UpperCamelCase     :Optional[Any]			  =		torch_device in ['''cpu''', '''mps''']

        self._test_inference_batch_single_identical(test_max_difference=__lowercase)







    @unittest.skipIf(
        torch_device != '''cuda''' or not is_xformers_available() , reason='''XFormers attention is only available with CUDA and `xformers` installed''' , )
    def 			UpperCamelCase__   (	self)       ->	Union[str, Any]:
        self._test_xformers_attention_forwardGenerator_pass(test_max_difference=__lowercase)


@slow
@require_torch_gpu
class 	lowerCamelCase_     (   unittest.TestCase ):






    '''simple docstring'''






    def 			UpperCamelCase__   (	self)       ->	Union[str, Any]:
        # clean up the VRAM after each test
        super().tearDown()
        gc.collect()
        torch.cuda.empty_cache()







    def 			UpperCamelCase__   (	self)       ->	Union[str, Any]:
        __UpperCamelCase     :int			  =		load_image(
            '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/stable_unclip/turtle.png''')

        __UpperCamelCase     :Any			  =		load_numpy(
            '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/stable_unclip/stable_unclip_2_1_l_img2img_anime_turtle_fp16.npy''')

        __UpperCamelCase     :List[Any]			  =		StableUnCLIPImgaImgPipeline.from_pretrained(
            '''fusing/stable-unclip-2-1-l-img2img''' , torch_dtype=torch.floataa)
        pipe.to(__lowercase)
        pipe.set_progress_bar_config(disable=__lowercase)
        # stable unclip will oom when integration tests are run on a V100,
        # so turn on memory savings
        pipe.enable_attention_slicing()
        pipe.enable_sequential_cpu_offload()

        __UpperCamelCase     :int			  =		torch.Generator(device='''cpu''').manual_seed(0)
        __UpperCamelCase     :Dict			  =		pipe(__lowercase , '''anime turle''' , generator=__lowercase , output_type='''np''')

        __UpperCamelCase     :Dict			  =		output.images[0]

        assert image.shape == (768, 768, 3)

        assert_mean_pixel_difference(__lowercase , __lowercase)







    def 			UpperCamelCase__   (	self)       ->	List[str]:
        __UpperCamelCase     :Optional[Any]			  =		load_image(
            '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/stable_unclip/turtle.png''')

        __UpperCamelCase     :Any			  =		load_numpy(
            '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/stable_unclip/stable_unclip_2_1_h_img2img_anime_turtle_fp16.npy''')

        __UpperCamelCase     :Any			  =		StableUnCLIPImgaImgPipeline.from_pretrained(
            '''fusing/stable-unclip-2-1-h-img2img''' , torch_dtype=torch.floataa)
        pipe.to(__lowercase)
        pipe.set_progress_bar_config(disable=__lowercase)
        # stable unclip will oom when integration tests are run on a V100,
        # so turn on memory savings
        pipe.enable_attention_slicing()
        pipe.enable_sequential_cpu_offload()

        __UpperCamelCase     :int			  =		torch.Generator(device='''cpu''').manual_seed(0)
        __UpperCamelCase     :Optional[int]			  =		pipe(__lowercase , '''anime turle''' , generator=__lowercase , output_type='''np''')

        __UpperCamelCase     :List[Any]			  =		output.images[0]

        assert image.shape == (768, 768, 3)

        assert_mean_pixel_difference(__lowercase , __lowercase)







    def 			UpperCamelCase__   (	self)       ->	List[str]:
        __UpperCamelCase     :Dict			  =		load_image(
            '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/stable_unclip/turtle.png''')

        torch.cuda.empty_cache()
        torch.cuda.reset_max_memory_allocated()
        torch.cuda.reset_peak_memory_stats()

        __UpperCamelCase     :List[Any]			  =		StableUnCLIPImgaImgPipeline.from_pretrained(
            '''fusing/stable-unclip-2-1-h-img2img''' , torch_dtype=torch.floataa)
        __UpperCamelCase     :Union[str, Any]			  =		pipe.to(__lowercase)
        pipe.set_progress_bar_config(disable=__lowercase)
        pipe.enable_attention_slicing()
        pipe.enable_sequential_cpu_offload()

        __UpperCamelCase     :Optional[Any]			  =		pipe(
            __lowercase , '''anime turtle''' , num_inference_steps=2 , output_type='''np''' , )

        __UpperCamelCase     :int			  =		torch.cuda.max_memory_allocated()
        # make sure that less than 7 GB is allocated
        assert mem_bytes < 7 * 10**9





