


from ..utils import (
    OptionalDependencyNotAvailable,
    is_flax_available,
    is_invisible_watermark_available,
    is_k_diffusion_available,
    is_librosa_available,
    is_note_seq_available,
    is_onnx_available,
    is_torch_available,
    is_transformers_available,
)


try:
				if not is_torch_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_pt_objects import *  # noqa F403
else:
				from .consistency_models import ConsistencyModelPipeline
				from .dance_diffusion import DanceDiffusionPipeline
				from .ddim import DDIMPipeline
				from .ddpm import DDPMPipeline
				from .dit import DiTPipeline
				from .latent_diffusion import LDMSuperResolutionPipeline
				from .latent_diffusion_uncond import LDMPipeline
				from .pipeline_utils import AudioPipelineOutput, DiffusionPipeline, ImagePipelineOutput
				from .pndm import PNDMPipeline
				from .repaint import RePaintPipeline
				from .score_sde_ve import ScoreSdeVePipeline
				from .stochastic_karras_ve import KarrasVePipeline

try:
				if not (is_torch_available() and is_librosa_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_torch_and_librosa_objects import *  # noqa F403
else:
				from .audio_diffusion import AudioDiffusionPipeline, Mel

try:
				if not (is_torch_available() and is_transformers_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_torch_and_transformers_objects import *  # noqa F403
else:
				from .alt_diffusion import AltDiffusionImgaImgPipeline, AltDiffusionPipeline
				from .audioldm import AudioLDMPipeline
				from .controlnet import (
				    StableDiffusionControlNetImgaImgPipeline,
				    StableDiffusionControlNetInpaintPipeline,
				    StableDiffusionControlNetPipeline,
				)
				from .deepfloyd_if import (
				    IFImgaImgPipeline,
				    IFImgaImgSuperResolutionPipeline,
				    IFInpaintingPipeline,
				    IFInpaintingSuperResolutionPipeline,
				    IFPipeline,
				    IFSuperResolutionPipeline,
				)
				from .kandinsky import (
				    KandinskyImgaImgPipeline,
				    KandinskyInpaintPipeline,
				    KandinskyPipeline,
				    KandinskyPriorPipeline,
				)
				from .kandinskya_a import (
				    KandinskyVaaControlnetImgaImgPipeline,
				    KandinskyVaaControlnetPipeline,
				    KandinskyVaaImgaImgPipeline,
				    KandinskyVaaInpaintPipeline,
				    KandinskyVaaPipeline,
				    KandinskyVaaPriorEmbaEmbPipeline,
				    KandinskyVaaPriorPipeline,
				)
				from .latent_diffusion import LDMTextToImagePipeline
				from .paint_by_example import PaintByExamplePipeline
				from .semantic_stable_diffusion import SemanticStableDiffusionPipeline
				from .shap_e import ShapEImgaImgPipeline, ShapEPipeline
				from .stable_diffusion import (
				    CycleDiffusionPipeline,
				    StableDiffusionAttendAndExcitePipeline,
				    StableDiffusionDepthaImgPipeline,
				    StableDiffusionDiffEditPipeline,
				    StableDiffusionImageVariationPipeline,
				    StableDiffusionImgaImgPipeline,
				    StableDiffusionInpaintPipeline,
				    StableDiffusionInpaintPipelineLegacy,
				    StableDiffusionInstructPixaPixPipeline,
				    StableDiffusionLatentUpscalePipeline,
				    StableDiffusionLDMaDPipeline,
				    StableDiffusionModelEditingPipeline,
				    StableDiffusionPanoramaPipeline,
				    StableDiffusionParadigmsPipeline,
				    StableDiffusionPipeline,
				    StableDiffusionPixaPixZeroPipeline,
				    StableDiffusionSAGPipeline,
				    StableDiffusionUpscalePipeline,
				    StableUnCLIPImgaImgPipeline,
				    StableUnCLIPPipeline,
				)
				from .stable_diffusion_safe import StableDiffusionPipelineSafe
				from .text_to_video_synthesis import TextToVideoSDPipeline, TextToVideoZeroPipeline, VideoToVideoSDPipeline
				from .unclip import UnCLIPImageVariationPipeline, UnCLIPPipeline
				from .unidiffuser import ImageTextPipelineOutput, UniDiffuserModel, UniDiffuserPipeline, UniDiffuserTextDecoder
				from .versatile_diffusion import (
				    VersatileDiffusionDualGuidedPipeline,
				    VersatileDiffusionImageVariationPipeline,
				    VersatileDiffusionPipeline,
				    VersatileDiffusionTextToImagePipeline,
				)
				from .vq_diffusion import VQDiffusionPipeline


try:
				if not (is_torch_available() and is_transformers_available() and is_invisible_watermark_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_torch_and_transformers_and_invisible_watermark_objects import *  # noqa F403
else:
				from .stable_diffusion_xl import StableDiffusionXLImgaImgPipeline, StableDiffusionXLPipeline

try:
				if not is_onnx_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_onnx_objects import *  # noqa F403
else:
				from .onnx_utils import OnnxRuntimeModel

try:
				if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_torch_and_transformers_and_onnx_objects import *  # noqa F403
else:
				from .stable_diffusion import (
				    OnnxStableDiffusionImgaImgPipeline,
				    OnnxStableDiffusionInpaintPipeline,
				    OnnxStableDiffusionInpaintPipelineLegacy,
				    OnnxStableDiffusionPipeline,
				    OnnxStableDiffusionUpscalePipeline,
				    StableDiffusionOnnxPipeline,
				)

try:
				if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_torch_and_transformers_and_k_diffusion_objects import *  # noqa F403
else:
				from .stable_diffusion import StableDiffusionKDiffusionPipeline

try:
				if not is_flax_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_flax_objects import *  # noqa F403
else:
				from .pipeline_flax_utils import FlaxDiffusionPipeline


try:
				if not (is_flax_available() and is_transformers_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_flax_and_transformers_objects import *  # noqa F403
else:
				from .controlnet import FlaxStableDiffusionControlNetPipeline
				from .stable_diffusion import (
				    FlaxStableDiffusionImgaImgPipeline,
				    FlaxStableDiffusionInpaintPipeline,
				    FlaxStableDiffusionPipeline,
				)
try:
				if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ..utils.dummy_transformers_and_torch_and_note_seq_objects import *  # noqa F403
else:
				from .spectrogram_diffusion import MidiProcessor, SpectrogramDiffusionPipeline













from typing import Dict, List, Optional, Union

import numpy as np

from ...image_processing_utils import BaseImageProcessor, BatchFeature, get_size_dict
from ...image_transforms import center_crop, normalize, rescale, resize, to_channel_dimension_format
from ...image_utils import (
    IMAGENET_DEFAULT_MEAN,
    IMAGENET_DEFAULT_STD,
    ChannelDimension,
    ImageInput,
    PILImageResampling,
    get_image_size,
    make_list_of_images,
    to_numpy_array,
    valid_images,
)
from ...utils import TensorType, is_vision_available, logging


if is_vision_available():
		import PIL


lowerCamelCase_						     =  logging.get_logger(__name__)




class 							a_ (						a_		):


							'''simple docstring'''






							__a:	Optional[Any]     =    ['''pixel_values''']

							def __init__(		self			,			lowercase_ = True			,			lowercase_ = None			,			lowercase_ = True			,			lowercase_ = None			,			lowercase_ = PILImageResampling.BICUBIC			,			lowercase_ = True			,			lowercase_ = 1 / 2_5_5			,			lowercase_ = True			,			lowercase_ = None			,			lowercase_ = None			,			**lowercase_			,			)					->					None:
														'''simple docstring'''

														super().__init__(**lowercase_   )
														lowerCAmelCase_										=     crop_size if crop_size is not None else {'height': 2_5_6, 'width': 2_5_6}
														lowerCAmelCase_										=     get_size_dict(lowercase_			,			param_name='crop_size'   )
														lowerCAmelCase_										=     size if size is not None else {'height': 2_2_4, 'width': 2_2_4}
														lowerCAmelCase_										=     get_size_dict(lowercase_   )

														lowerCAmelCase_										=     do_center_crop
														lowerCAmelCase_										=     crop_size
														lowerCAmelCase_										=     do_resize
														lowerCAmelCase_										=     size
														lowerCAmelCase_										=     resample
														lowerCAmelCase_										=     do_rescale
														lowerCAmelCase_										=     rescale_factor
														lowerCAmelCase_										=     do_normalize
														lowerCAmelCase_										=     image_mean if image_mean is not None else IMAGENET_DEFAULT_MEAN
														lowerCAmelCase_										=     image_std if image_std is not None else IMAGENET_DEFAULT_STD

							def      _lowercase       (		self			,			lowercase_			,			lowercase_			,			lowercase_ = None			,			lowercase_ = None			,			**lowercase_			,			)					->					np.ndarray:
														'''simple docstring'''

														lowerCAmelCase_										=     self.size if size is None else size
														lowerCAmelCase_										=     get_size_dict(lowercase_   )
														lowerCAmelCase_										=     get_size_dict(lowercase_			,			param_name='crop_size'   )

														lowerCAmelCase_						,   lowerCAmelCase_										=     get_image_size(lowercase_   )
														lowerCAmelCase_										=     min(lowercase_			,			lowercase_   )
														lowerCAmelCase_										=     (size['height'] / crop_size['height']) * min_dim
														lowerCAmelCase_										=     (size['width'] / crop_size['width']) * min_dim
														return center_crop(lowercase_			,			size=(cropped_height, cropped_width)			,			data_format=lowercase_			,			**lowercase_   )

							def      _lowercase       (		self			,			lowercase_			,			lowercase_			,			lowercase_ = PIL.Image.BICUBIC			,			lowercase_ = None			,			**lowercase_			,			)					->					np.ndarray:
														'''simple docstring'''

														lowerCAmelCase_										=     get_size_dict(lowercase_   )
														if "height" not in size or "width" not in size:
																					raise ValueError(f'''The size dictionary must contain the keys \'height\' and \'width\'. Got {size.keys()}'''   )
														return resize(
														    lowercase_			,			size=(size['height'], size['width'])			,			resample=lowercase_			,			data_format=lowercase_			,			**lowercase_   )

							def      _lowercase       (		self			,			lowercase_			,			lowercase_			,			lowercase_ = None			,			**lowercase_			,			)					->					Any:
														'''simple docstring'''

														return rescale(lowercase_			,			scale=lowercase_			,			data_format=lowercase_			,			**lowercase_   )

							def      _lowercase       (		self			,			lowercase_			,			lowercase_			,			lowercase_			,			lowercase_ = None			,			**lowercase_			,			)					->					np.ndarray:
														'''simple docstring'''

														return normalize(lowercase_			,			mean=lowercase_			,			std=lowercase_			,			data_format=lowercase_			,			**lowercase_   )






							def      _lowercase       (		self			,			lowercase_			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = None			,			lowercase_ = ChannelDimension.FIRST			,			**lowercase_			,			)					->					PIL.Image.Image:
														'''simple docstring'''

														lowerCAmelCase_										=     do_center_crop if do_center_crop is not None else self.do_center_crop
														lowerCAmelCase_										=     crop_size if crop_size is not None else self.crop_size
														lowerCAmelCase_										=     get_size_dict(lowercase_			,			param_name='crop_size'   )
														lowerCAmelCase_										=     do_resize if do_resize is not None else self.do_resize
														lowerCAmelCase_										=     size if size is not None else self.size
														lowerCAmelCase_										=     get_size_dict(lowercase_   )
														lowerCAmelCase_										=     resample if resample is not None else self.resample
														lowerCAmelCase_										=     do_rescale if do_rescale is not None else self.do_rescale
														lowerCAmelCase_										=     rescale_factor if rescale_factor is not None else self.rescale_factor
														lowerCAmelCase_										=     do_normalize if do_normalize is not None else self.do_normalize
														lowerCAmelCase_										=     image_mean if image_mean is not None else self.image_mean
														lowerCAmelCase_										=     image_std if image_std is not None else self.image_std

														lowerCAmelCase_										=     make_list_of_images(lowercase_   )

														if not valid_images(lowercase_   ):
																					raise ValueError(
																					    'Invalid image type. Must be of type PIL.Image.Image, numpy.ndarray, '
																					    'torch.Tensor, tf.Tensor or jax.ndarray.'   )

														if do_center_crop and crop_size is None:
																					raise ValueError('If `do_center_crop` is set to `True`, `crop_size` must be provided.'   )

														if do_resize and size is None:
																					raise ValueError('Size must be specified if do_resize is True.'   )

														if do_rescale and rescale_factor is None:
																					raise ValueError('Rescale factor must be specified if do_rescale is True.'   )

														if do_normalize and (image_mean is None or image_std is None):
																					raise ValueError('Image mean and image standard deviation must be specified if do_normalize is True.'   )

														# All transformations expect numpy arrays.
														lowerCAmelCase_										=     [to_numpy_array(lowercase_   ) for image in images]

														if do_center_crop:
																					lowerCAmelCase_										=     [self.center_crop(lowercase_			,			lowercase_			,			size=lowercase_   ) for image in images]

														if do_resize:
																					lowerCAmelCase_										=     [self.resize(image=lowercase_			,			size=lowercase_			,			resample=lowercase_   ) for image in images]

														if do_rescale:
																					lowerCAmelCase_										=     [self.rescale(image=lowercase_			,			scale=lowercase_   ) for image in images]

														if do_normalize:
																					lowerCAmelCase_										=     [self.normalize(image=lowercase_			,			mean=lowercase_			,			std=lowercase_   ) for image in images]

														lowerCAmelCase_										=     [to_channel_dimension_format(lowercase_			,			lowercase_   ) for image in images]

														lowerCAmelCase_										=     {'pixel_values': images}
														return BatchFeature(data=lowercase_			,			tensor_type=lowercase_   )



