






from typing import TYPE_CHECKING

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


lowerCamelCase	:							Tuple				={
    '''configuration_altclip''': [
        '''ALTCLIP_PRETRAINED_CONFIG_ARCHIVE_MAP''',
        '''AltCLIPConfig''',
        '''AltCLIPTextConfig''',
        '''AltCLIPVisionConfig''',
    ],
    '''processing_altclip''': ['''AltCLIPProcessor'''],
}

try:
 if not is_torch_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 lowerCamelCase	:							Optional[int]				=[
     '''ALTCLIP_PRETRAINED_MODEL_ARCHIVE_LIST''',
     '''AltCLIPPreTrainedModel''',
     '''AltCLIPModel''',
     '''AltCLIPTextModel''',
     '''AltCLIPVisionModel''',
 ]


if TYPE_CHECKING:
 from .configuration_altclip import (
     ALTCLIP_PRETRAINED_CONFIG_ARCHIVE_MAP,
     AltCLIPConfig,
     AltCLIPTextConfig,
     AltCLIPVisionConfig,
 )
 from .processing_altclip import AltCLIPProcessor

 try:
  if not is_torch_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .modeling_altclip import (
      ALTCLIP_PRETRAINED_MODEL_ARCHIVE_LIST,
      AltCLIPModel,
      AltCLIPPreTrainedModel,
      AltCLIPTextModel,
      AltCLIPVisionModel,
  )


else:
 import sys

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







import unittest

import numpy as np
import torch

from diffusers import VersatileDiffusionImageVariationPipeline
from diffusers.utils.testing_utils import load_image, require_torch_gpu, slow, torch_device


lowerCamelCase	:							Tuple				=False





class 			__a  (    unittest.TestCase       ):
 pass







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







 def 					__lowercase					(		self					:   int						):


      '''simple docstring'''






      UpperCamelCase__			:   Optional[Any]       				=  VersatileDiffusionImageVariationPipeline.from_pretrained("shi-labs/versatile-diffusion"						)
      pipe.to(SCREAMING_SNAKE_CASE						)
      pipe.set_progress_bar_config(disable=SCREAMING_SNAKE_CASE						)

      UpperCamelCase__			:   Union[str, Any]       				=  load_image(
          "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg"						)
      UpperCamelCase__			:   List[Any]       				=  torch.manual_seed(0						)
      UpperCamelCase__			:   List[Any]       				=  pipe(
          image=SCREAMING_SNAKE_CASE							,   generator=SCREAMING_SNAKE_CASE							,   guidance_scale=7.5							,   num_inference_steps=50							,   output_type="numpy"							,   ).images

      UpperCamelCase__			:   Union[str, Any]       				=  image[0, 2_53:2_56, 2_53:2_56, -1]

      assert image.shape == (1, 5_12, 5_12, 3)
      UpperCamelCase__			:   Optional[Any]       				=  np.array([0.0_4_4_1, 0.0_4_6_9, 0.0_5_0_7, 0.0_5_7_5, 0.0_6_3_2, 0.0_6_5_0, 0.0_8_6_5, 0.0_9_0_9, 0.0_9_4_5]						)

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