



'''simple docstring'''







import gc
import tempfile
import unittest

import numpy as np
import torch

from diffusers import VersatileDiffusionDualGuidedPipeline
from diffusers.utils.testing_utils import load_image, nightly, require_torch_gpu, torch_device


SCREAMING_SNAKE_CASE__     : str				=   False




@nightly
@require_torch_gpu
class 				__lowerCAmelCase(							unittest.TestCase      ):






			def 				_lowercase	(   self  :							List[Any]	):




					"""simple docstring"""







					super().tearDown()
					gc.collect()
					torch.cuda.empty_cache()






			def 				_lowercase	(   self  :							Any	):




					"""simple docstring"""







					SCREAMING_SNAKE_CASE_						:Dict										=      VersatileDiffusionDualGuidedPipeline.from_pretrained('shi-labs/versatile-diffusion'	)
					# remove text_unet
					pipe.remove_unused_weights()
					pipe.to(snake_case__	)
					pipe.set_progress_bar_config(disable=snake_case__	)

					SCREAMING_SNAKE_CASE_						:Optional[Any]										=      load_image(
					    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg'	)

					SCREAMING_SNAKE_CASE_						:Union[str, Any]										=      torch.manual_seed(0	)
					SCREAMING_SNAKE_CASE_						:Tuple										=      pipe(
					    prompt='first prompt'      ,       image=snake_case__      ,       text_to_image_strength=0.75      ,       generator=snake_case__      ,       guidance_scale=7.5      ,       num_inference_steps=2      ,       output_type='numpy'      ,       ).images

					with tempfile.TemporaryDirectory() as tmpdirname:
							pipe.save_pretrained(snake_case__	)
							SCREAMING_SNAKE_CASE_						:str										=      VersatileDiffusionDualGuidedPipeline.from_pretrained(snake_case__	)

					pipe.to(snake_case__	)
					pipe.set_progress_bar_config(disable=snake_case__	)

					SCREAMING_SNAKE_CASE_						:List[str]										=      generator.manual_seed(0	)
					SCREAMING_SNAKE_CASE_						:Optional[Any]										=      pipe(
					    prompt='first prompt'      ,       image=snake_case__      ,       text_to_image_strength=0.75      ,       generator=snake_case__      ,       guidance_scale=7.5      ,       num_inference_steps=2      ,       output_type='numpy'      ,       ).images

					assert np.abs(image - new_image	).sum() < 1E-5, "Models don't have the same forward pass"






			def 				_lowercase	(   self  :							str	):




					"""simple docstring"""







					SCREAMING_SNAKE_CASE_						:str										=      VersatileDiffusionDualGuidedPipeline.from_pretrained('shi-labs/versatile-diffusion'	)
					pipe.remove_unused_weights()
					pipe.to(snake_case__	)
					pipe.set_progress_bar_config(disable=snake_case__	)

					SCREAMING_SNAKE_CASE_						:List[Any]										=      'cyberpunk 2077'
					SCREAMING_SNAKE_CASE_						:Any										=      load_image(
					    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg'	)
					SCREAMING_SNAKE_CASE_						:List[str]										=      torch.manual_seed(0	)
					SCREAMING_SNAKE_CASE_						:Dict										=      pipe(
					    prompt=snake_case__      ,       image=snake_case__      ,       text_to_image_strength=0.75      ,       generator=snake_case__      ,       guidance_scale=7.5      ,       num_inference_steps=50      ,       output_type='numpy'      ,       ).images

					SCREAMING_SNAKE_CASE_						:Optional[int]										=      image[0, 253:256, 253:256, -1]

					assert image.shape == (1, 512, 512, 3)
					SCREAMING_SNAKE_CASE_						:Optional[Any]										=      np.array([0.07_87, 0.08_49, 0.08_26, 0.08_12, 0.08_07, 0.07_95, 0.08_18, 0.07_98, 0.07_79]	)

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








import logging
import os
import sys
import threading
from logging import (
    CRITICAL,  # NOQA
    DEBUG,  # NOQA
    ERROR,  # NOQA
    FATAL,  # NOQA
    INFO,  # NOQA
    NOTSET,  # NOQA
    WARN,  # NOQA
    WARNING,  # NOQA
)
from typing import Optional

from tqdm import auto as tqdm_lib


lowerCamelCase					    =	threading.Lock()
lowerCamelCase					    =	None

lowerCamelCase					    =	{
    '''debug''': logging.DEBUG,
    '''info''': logging.INFO,
    '''warning''': logging.WARNING,
    '''error''': logging.ERROR,
    '''critical''': logging.CRITICAL,
}

lowerCamelCase					    =	logging.WARNING

lowerCamelCase					    =	True
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _lowercase     			=						os.getenv('DIFFUSERS_VERBOSITY',   _lowercase   )
    if env_level_str:
        if env_level_str in log_levels:
            return log_levels[env_level_str]
        else:
            logging.getLogger().warning(
                f'''Unknown option DIFFUSERS_VERBOSITY={env_level_str}, '''
                f'''has to be one of: { ", ".join(log_levels.keys()   ) }'''   )
    return _default_log_level
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    return __name__.split('.'   )[0]
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    return logging.getLogger(_get_library_name()   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    global _default_handler

    with _lock:
        if _default_handler:
            # This library has already configured the library root logger.
            return
        _lowercase     			=						logging.StreamHandler()  # Set sys.stderr as stream.
        _lowercase     			=						sys.stderr.flush

        # Apply our default configuration to the library root logger.
        _lowercase     			=						_get_library_root_logger()
        library_root_logger.addHandler(_default_handler   )
        library_root_logger.setLevel(_get_default_logging_level()   )
        _lowercase     			=						False
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    global _default_handler

    with _lock:
        if not _default_handler:
            return

        _lowercase     			=						_get_library_root_logger()
        library_root_logger.removeHandler(_default_handler   )
        library_root_logger.setLevel(logging.NOTSET   )
        _lowercase     			=						None
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    return log_levels
def 							UpperCAmelCase   ( _lowercase = None   ):


    '''simple docstring'''



    if name is None:
        _lowercase     			=						_get_library_name()

    _configure_library_root_logger()
    return logging.getLogger(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _configure_library_root_logger()
    return _get_library_root_logger().getEffectiveLevel()
def 							UpperCAmelCase   ( _lowercase   ):


    '''simple docstring'''



    _configure_library_root_logger()
    _get_library_root_logger().setLevel(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    return set_verbosity(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    return set_verbosity(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    return set_verbosity(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    return set_verbosity(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _configure_library_root_logger()

    assert _default_handler is not None
    _get_library_root_logger().removeHandler(_default_handler   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _configure_library_root_logger()

    assert _default_handler is not None
    _get_library_root_logger().addHandler(_default_handler   )
def 							UpperCAmelCase   ( _lowercase   ):


    '''simple docstring'''



    _configure_library_root_logger()

    assert handler is not None
    _get_library_root_logger().addHandler(_lowercase   )
def 							UpperCAmelCase   ( _lowercase   ):


    '''simple docstring'''



    _configure_library_root_logger()

    assert handler is not None and handler not in _get_library_root_logger().handlers
    _get_library_root_logger().removeHandler(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _configure_library_root_logger()
    _lowercase     			=						False
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _configure_library_root_logger()
    _lowercase     			=						True
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _lowercase     			=						_get_library_root_logger().handlers

    for handler in handlers:
        _lowercase     			=						logging.Formatter('[%(levelname)s|%(filename)s:%(lineno)s] %(asctime)s >> %(message)s'   )
        handler.setFormatter(_lowercase   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    _lowercase     			=						_get_library_root_logger().handlers

    for handler in handlers:
        handler.setFormatter(_lowercase   )
def 							UpperCAmelCase   ( self,   *_lowercase,   **_lowercase   ):


    '''simple docstring'''



    _lowercase     			=						os.getenv('DIFFUSERS_NO_ADVISORY_WARNINGS',   _lowercase   )
    if no_advisory_warnings:
        return
    self.warning(*_lowercase,   **_lowercase   )


lowerCamelCase					    =	warning_advice

class        snake_case_				:







   """simple docstring"""



   def __init__(     self:						int ,							*snake_case__:						str ,							**snake_case__:						Any						)	->		Any:  # pylint: disable=unused-argument


       """simple docstring"""



       _lowercase     			=						args[0] if args else None

   def __iter__(     self:						Union[str, Any]						)	->		str:


       """simple docstring"""



       return iter(self._iterator						)

   def __getattr__(     self:						int ,							snake_case__:						Any						)	->		List[Any]:


       """simple docstring"""



       def empty_fn(*snake_case__:						Optional[int] ,							**snake_case__:						Union[str, Any]						):  # pylint: disable=unused-argument
           return

       return empty_fn

   def __enter__(     self:						List[Any]						)	->		Tuple:


       """simple docstring"""



       return self

   def __exit__(     self:						List[str] ,							snake_case__:						Optional[Any] ,							snake_case__:						Dict ,							snake_case__:						List[str]						)	->		Optional[Any]:


       """simple docstring"""



       return






class        snake_case_				:







   """simple docstring"""



   def __call__(     self:						List[str] ,							*snake_case__:						Tuple ,							**snake_case__:						Union[str, Any]						)	->		List[str]:


       """simple docstring"""



       if _tqdm_active:
           return tqdm_lib.tqdm(*snake_case__ ,							**snake_case__						)
       else:
           return EmptyTqdm(*snake_case__ ,							**snake_case__						)

   def       SCREAMING_SNAKE_CASE__					(     self:						Tuple ,							*snake_case__:						Union[str, Any] ,							**snake_case__:						str						)	->		Optional[int]:


       """simple docstring"""



       _lowercase     			=						None
       if _tqdm_active:
           return tqdm_lib.tqdm.set_lock(*snake_case__ ,							**snake_case__						)

   def       SCREAMING_SNAKE_CASE__					(     self:						int						)	->		int:


       """simple docstring"""



       if _tqdm_active:
           return tqdm_lib.tqdm.get_lock()






lowerCamelCase					    =	_tqdm_cls()
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    global _tqdm_active
    return bool(_tqdm_active   )
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    global _tqdm_active
    _lowercase     			=						True
def 							UpperCAmelCase   ( ):


    '''simple docstring'''



    global _tqdm_active
    _lowercase     			=						False
