from collections import OrderedDict
from typing import List, Mapping

from packaging import version

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


a  :      Optional[int]            =    logging.get_logger(__name__)

a  :      Optional[Any]            =    {
    """google/efficientnet-b7""": """https://huggingface.co/google/efficientnet-b7/resolve/main/config.json""",
}

class 					_lowercase    (     UpperCAmelCase__			):






    '''simple docstring'''
    SCREAMING_SNAKE_CASE:				Dict          =    'efficientnet'




    def __init__(    self       ,						lowerCamelCase__ = 3       ,						lowerCamelCase__ = 600       ,						lowerCamelCase__ = 2.0       ,						lowerCamelCase__ = 3.1       ,						lowerCamelCase__ = 8       ,						lowerCamelCase__ = [3, 3, 5, 3, 5, 5, 3]       ,						lowerCamelCase__ = [32, 16, 24, 40, 80, 112, 192]       ,						lowerCamelCase__ = [16, 24, 40, 80, 112, 192, 320]       ,						lowerCamelCase__ = []       ,						lowerCamelCase__ = [1, 2, 2, 2, 1, 2, 1]       ,						lowerCamelCase__ = [1, 2, 2, 3, 3, 4, 1]       ,						lowerCamelCase__ = [1, 6, 6, 6, 6, 6, 6]       ,						lowerCamelCase__ = 0.2_5       ,						lowerCamelCase__ = "swish"       ,						lowerCamelCase__ = 2_560       ,						lowerCamelCase__ = "mean"       ,						lowerCamelCase__ = 0.0_2       ,						lowerCamelCase__ = 0.0_0_1       ,						lowerCamelCase__ = 0.9_9       ,						lowerCamelCase__ = 0.5       ,						lowerCamelCase__ = 0.2       ,						**lowerCamelCase__       ,						):
       super().__init__(**lowerCamelCase__ )

       lowerCAmelCase_:							str       =	num_channels
       lowerCAmelCase_:							str       =	image_size
       lowerCAmelCase_:							int       =	width_coefficient
       lowerCAmelCase_:							Union[str, Any]       =	depth_coefficient
       lowerCAmelCase_:							int       =	depth_divisor
       lowerCAmelCase_:							List[str]       =	kernel_sizes
       lowerCAmelCase_:							Tuple       =	in_channels
       lowerCAmelCase_:							List[str]       =	out_channels
       lowerCAmelCase_:							List[str]       =	depthwise_padding
       lowerCAmelCase_:							Optional[int]       =	strides
       lowerCAmelCase_:							List[str]       =	num_block_repeats
       lowerCAmelCase_:							Any       =	expand_ratios
       lowerCAmelCase_:							List[Any]       =	squeeze_expansion_ratio
       lowerCAmelCase_:							Optional[int]       =	hidden_act
       lowerCAmelCase_:							Optional[int]       =	hidden_dim
       lowerCAmelCase_:							Dict       =	pooling_type
       lowerCAmelCase_:							Optional[Any]       =	initializer_range
       lowerCAmelCase_:							int       =	batch_norm_eps
       lowerCAmelCase_:							List[str]       =	batch_norm_momentum
       lowerCAmelCase_:							List[Any]       =	dropout_rate
       lowerCAmelCase_:							Union[str, Any]       =	drop_connect_rate
       lowerCAmelCase_:							Tuple       =	sum(lowerCamelCase__ ) * 4





class 					_lowercase    (     UpperCAmelCase__			):






    '''simple docstring'''
    SCREAMING_SNAKE_CASE:				Optional[Any]          =    version.parse('1.11'			)




    @property
    def      _a    (    self ):
       return OrderedDict(
           [
               ("pixel_values", {0: "batch", 1: "num_channels", 2: "height", 3: "width"}),
           ] )




    @property
    def      _a    (    self ):
       return 1E-5
import gc
import tempfile
import unittest

import numpy as np
import torch

from diffusers import VersatileDiffusionTextToImagePipeline
from diffusers.utils.testing_utils import nightly, require_torch_gpu, torch_device


a  :      List[str]            =    False

class 					_lowercase    (     unittest.TestCase			):






    '''simple docstring'''
    pass





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






    '''simple docstring'''




    def      _a    (    self ):
       # clean up the VRAM after each test
       super().tearDown()
       gc.collect()
       torch.cuda.empty_cache()




    def      _a    (    self ):
       lowerCAmelCase_:							List[str]       =	VersatileDiffusionTextToImagePipeline.from_pretrained("shi-labs/versatile-diffusion" )
       # remove text_unet
       pipe.remove_unused_weights()
       pipe.to(lowerCamelCase__ )
       pipe.set_progress_bar_config(disable=lowerCamelCase__ )

       lowerCAmelCase_:							Tuple       =	"A painting of a squirrel eating a burger "
       lowerCAmelCase_:							List[str]       =	torch.manual_seed(0 )
       lowerCAmelCase_:							Dict       =	pipe(
           prompt=lowerCamelCase__       ,						generator=lowerCamelCase__       ,						guidance_scale=7.5       ,						num_inference_steps=2       ,						output_type="numpy" ).images

       with tempfile.TemporaryDirectory() as tmpdirname:
          pipe.save_pretrained(lowerCamelCase__ )
          lowerCAmelCase_:							int       =	VersatileDiffusionTextToImagePipeline.from_pretrained(lowerCamelCase__ )
       pipe.to(lowerCamelCase__ )
       pipe.set_progress_bar_config(disable=lowerCamelCase__ )

       lowerCAmelCase_:							List[str]       =	generator.manual_seed(0 )
       lowerCAmelCase_:							Dict       =	pipe(
           prompt=lowerCamelCase__       ,						generator=lowerCamelCase__       ,						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      _a    (    self ):
       lowerCAmelCase_:							Optional[int]       =	VersatileDiffusionTextToImagePipeline.from_pretrained(
           "shi-labs/versatile-diffusion"       ,						torch_dtype=torch.floataa )
       pipe.to(lowerCamelCase__ )
       pipe.set_progress_bar_config(disable=lowerCamelCase__ )

       lowerCAmelCase_:							str       =	"A painting of a squirrel eating a burger "
       lowerCAmelCase_:							int       =	torch.manual_seed(0 )
       lowerCAmelCase_:							List[str]       =	pipe(
           prompt=lowerCamelCase__       ,						generator=lowerCamelCase__       ,						guidance_scale=7.5       ,						num_inference_steps=50       ,						output_type="numpy" ).images

       lowerCAmelCase_:							List[Any]       =	image[0, 253:256, 253:256, -1]

       assert image.shape == (1, 512, 512, 3)
       lowerCAmelCase_:							Optional[int]       =	np.array([0.3_3_6_7, 0.3_1_6_9, 0.2_6_5_6, 0.3_8_7_0, 0.4_7_9_0, 0.3_7_9_6, 0.4_0_0_9, 0.4_8_7_8, 0.4_7_7_8] )

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