

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









class     SCREAMING_SNAKE_CASE__     :





       '''simple docstring'''




       def __init__(						self       :						Tuple							)			->    None:
             # numbermap keys are the number and its values are lists of indexes sorted
             # in ascending order
             snake_case__					=   {}
             # indexmap keys are an index and it's values are the number at that index
             snake_case__					=   {}

       def UpperCAmelCase_						(						self       :						Optional[int]    ,		lowerCAmelCase__       :						list | str | range    ,		lowerCAmelCase__       :						int							)			->    list[int]:
             if isinstance(lowerCAmelCase__    ,		(range, str)							):
                   snake_case__					=   list(lowerCAmelCase__							)
             elif not isinstance(lowerCAmelCase__    ,		lowerCAmelCase__							):
                   raise TypeError(
                       """binary_search_delete() only accepts either a list, range or str"""							)

             snake_case__					=   0
             snake_case__					=   len(lowerCAmelCase__							) - 1

             while low <= high:
                   snake_case__					=   (low + high) // 2
                   if array[mid] == item:
                         array.pop(lowerCAmelCase__							)
                         return array
                   elif array[mid] < item:
                         snake_case__					=   mid + 1
                   else:
                         snake_case__					=   mid - 1
             raise ValueError(
                 """Either the item is not in the array or the array was unsorted"""							)

       def UpperCAmelCase_						(						self       :						Optional[Any]    ,		lowerCAmelCase__       :						list | str | range    ,		lowerCAmelCase__       :						int							)			->    list[int]:
             if isinstance(lowerCAmelCase__    ,		(range, str)							):
                   snake_case__					=   list(lowerCAmelCase__							)
             elif not isinstance(lowerCAmelCase__    ,		lowerCAmelCase__							):
                   raise TypeError(
                       """binary_search_insert() only accepts either a list, range or str"""							)

             snake_case__					=   0
             snake_case__					=   len(lowerCAmelCase__							) - 1

             while low <= high:
                   snake_case__					=   (low + high) // 2
                   if array[mid] == index:
                         # If the item already exists in the array,
                         # insert it after the existing item
                         array.insert(mid + 1    ,		lowerCAmelCase__							)
                         return array
                   elif array[mid] < index:
                         snake_case__					=   mid + 1
                   else:
                         snake_case__					=   mid - 1

        # If the item doesn't exist in the array, insert it at the appropriate position
             array.insert(lowerCAmelCase__    ,		lowerCAmelCase__							)
             return array

       def UpperCAmelCase_						(						self       :						str    ,		lowerCAmelCase__       :						int    ,		lowerCAmelCase__       :						int							)			->    None:
             # Remove previous index
             if index in self.indexmap:
                   snake_case__					=   self.indexmap[index]
                   if len(self.numbermap[n]							) == 1:
                         del self.numbermap[n]
                   else:
                         snake_case__					=   self.binary_search_delete(self.numbermap[n]    ,		lowerCAmelCase__							)

        # Set new index
             snake_case__					=   number

             # Number not seen before or empty so insert number value
             if number not in self.numbermap:
                   snake_case__					=   [index]

             # Here we need to perform a binary search insertion in order to insert
             # The item in the correct place
             else:
                   snake_case__					=   self.binary_search_insert(
                       self.numbermap[number]    ,		lowerCAmelCase__							)

       def UpperCAmelCase_						(						self       :						Any    ,		lowerCAmelCase__       :						int							)			->    int:
             return self.numbermap.get(lowerCAmelCase__    ,		[-1]							)[0]


if __name__ == "__main__":
       import doctest

       doctest.testmod()






