Ignore jupyter checkpoints

.gitignore

@@ -7,3 +7,4 @@ preprocess/mhp_extension/demo/*.jpg
 # blocked by HF binary hook
 preprocess/demo/*.jpg
 preprocess/mhp_extension/demo/*.jpg
+**/.ipynb_checkpoints/

diffusers3/.ipynb_checkpoints/__init__-checkpoint.py

@@ -1,934 +0,0 @@
-__version__ = "0.31.0.dev0"
-
-from typing import TYPE_CHECKING
-
-from .utils import (
-    DIFFUSERS_SLOW_IMPORT,
-    OptionalDependencyNotAvailable,
-    _LazyModule,
-    is_flax_available,
-    is_k_diffusion_available,
-    is_librosa_available,
-    is_note_seq_available,
-    is_onnx_available,
-    is_scipy_available,
-    is_sentencepiece_available,
-    is_torch_available,
-    is_torchsde_available,
-    is_transformers_available,
-)
-
-
-# Lazy Import based on
-# https://github.com/huggingface/transformers/blob/main/src/transformers/__init__.py
-
-# When adding a new object to this init, please add it to `_import_structure`. The `_import_structure` is a dictionary submodule to list of object names,
-# and is used to defer the actual importing for when the objects are requested.
-# This way `import diffusers` provides the names in the namespace without actually importing anything (and especially none of the backends).
-
-_import_structure = {
-    "configuration_utils": ["ConfigMixin"],
-    "loaders": ["FromOriginalModelMixin"],
-    "models": [],
-    "pipelines": [],
-    "schedulers": [],
-    "utils": [
-        "OptionalDependencyNotAvailable",
-        "is_flax_available",
-        "is_inflect_available",
-        "is_invisible_watermark_available",
-        "is_k_diffusion_available",
-        "is_k_diffusion_version",
-        "is_librosa_available",
-        "is_note_seq_available",
-        "is_onnx_available",
-        "is_scipy_available",
-        "is_torch_available",
-        "is_torchsde_available",
-        "is_transformers_available",
-        "is_transformers_version",
-        "is_unidecode_available",
-        "logging",
-    ],
-}
-
-try:
-    if not is_onnx_available():
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_onnx_objects  # noqa F403
-
-    _import_structure["utils.dummy_onnx_objects"] = [
-        name for name in dir(dummy_onnx_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["pipelines"].extend(["OnnxRuntimeModel"])
-
-try:
-    if not is_torch_available():
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_pt_objects  # noqa F403
-
-    _import_structure["utils.dummy_pt_objects"] = [name for name in dir(dummy_pt_objects) if not name.startswith("_")]
-
-else:
-    _import_structure["models"].extend(
-        [
-            "AsymmetricAutoencoderKL",
-            "AuraFlowTransformer2DModel",
-            "AutoencoderKL",
-            "AutoencoderKLCogVideoX",
-            "AutoencoderKLTemporalDecoder",
-            "AutoencoderOobleck",
-            "AutoencoderTiny",
-            "CogVideoXTransformer3DModel",
-            "ConsistencyDecoderVAE",
-            "ControlNetModel",
-            "ControlNetXSAdapter",
-            "DiTTransformer2DModel",
-            "FluxControlNetModel",
-            "FluxMultiControlNetModel",
-            "FluxTransformer2DModel",
-            "HunyuanDiT2DControlNetModel",
-            "HunyuanDiT2DModel",
-            "HunyuanDiT2DMultiControlNetModel",
-            "I2VGenXLUNet",
-            "Kandinsky3UNet",
-            "LatteTransformer3DModel",
-            "LuminaNextDiT2DModel",
-            "ModelMixin",
-            "MotionAdapter",
-            "MultiAdapter",
-            "PixArtTransformer2DModel",
-            "PriorTransformer",
-            "SD3ControlNetModel",
-            "SD3MultiControlNetModel",
-            "SD3Transformer2DModel",
-            "SparseControlNetModel",
-            "StableAudioDiTModel",
-            "StableCascadeUNet",
-            "T2IAdapter",
-            "T5FilmDecoder",
-            "Transformer2DModel",
-            "UNet1DModel",
-            "UNet2DConditionModel",
-            "UNet2DModel",
-            "UNet3DConditionModel",
-            "UNetControlNetXSModel",
-            "UNetMotionModel",
-            "UNetSpatioTemporalConditionModel",
-            "UVit2DModel",
-            "VQModel",
-        ]
-    )
-
-    _import_structure["optimization"] = [
-        "get_constant_schedule",
-        "get_constant_schedule_with_warmup",
-        "get_cosine_schedule_with_warmup",
-        "get_cosine_with_hard_restarts_schedule_with_warmup",
-        "get_linear_schedule_with_warmup",
-        "get_polynomial_decay_schedule_with_warmup",
-        "get_scheduler",
-    ]
-    _import_structure["pipelines"].extend(
-        [
-            "AudioPipelineOutput",
-            "AutoPipelineForImage2Image",
-            "AutoPipelineForInpainting",
-            "AutoPipelineForText2Image",
-            "ConsistencyModelPipeline",
-            "DanceDiffusionPipeline",
-            "DDIMPipeline",
-            "DDPMPipeline",
-            "DiffusionPipeline",
-            "DiTPipeline",
-            "ImagePipelineOutput",
-            "KarrasVePipeline",
-            "LDMPipeline",
-            "LDMSuperResolutionPipeline",
-            "PNDMPipeline",
-            "RePaintPipeline",
-            "ScoreSdeVePipeline",
-            "StableDiffusionMixin",
-        ]
-    )
-    _import_structure["schedulers"].extend(
-        [
-            "AmusedScheduler",
-            "CMStochasticIterativeScheduler",
-            "CogVideoXDDIMScheduler",
-            "CogVideoXDPMScheduler",
-            "DDIMInverseScheduler",
-            "DDIMParallelScheduler",
-            "DDIMScheduler",
-            "DDPMParallelScheduler",
-            "DDPMScheduler",
-            "DDPMWuerstchenScheduler",
-            "DEISMultistepScheduler",
-            "DPMSolverMultistepInverseScheduler",
-            "DPMSolverMultistepScheduler",
-            "DPMSolverSinglestepScheduler",
-            "EDMDPMSolverMultistepScheduler",
-            "EDMEulerScheduler",
-            "EulerAncestralDiscreteScheduler",
-            "EulerDiscreteScheduler",
-            "FlowMatchEulerDiscreteScheduler",
-            "FlowMatchHeunDiscreteScheduler",
-            "HeunDiscreteScheduler",
-            "IPNDMScheduler",
-            "KarrasVeScheduler",
-            "KDPM2AncestralDiscreteScheduler",
-            "KDPM2DiscreteScheduler",
-            "LCMScheduler",
-            "PNDMScheduler",
-            "RePaintScheduler",
-            "SASolverScheduler",
-            "SchedulerMixin",
-            "ScoreSdeVeScheduler",
-            "TCDScheduler",
-            "UnCLIPScheduler",
-            "UniPCMultistepScheduler",
-            "VQDiffusionScheduler",
-        ]
-    )
-    _import_structure["training_utils"] = ["EMAModel"]
-
-try:
-    if not (is_torch_available() and is_scipy_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_torch_and_scipy_objects  # noqa F403
-
-    _import_structure["utils.dummy_torch_and_scipy_objects"] = [
-        name for name in dir(dummy_torch_and_scipy_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["schedulers"].extend(["LMSDiscreteScheduler"])
-
-try:
-    if not (is_torch_available() and is_torchsde_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_torch_and_torchsde_objects  # noqa F403
-
-    _import_structure["utils.dummy_torch_and_torchsde_objects"] = [
-        name for name in dir(dummy_torch_and_torchsde_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["schedulers"].extend(["CosineDPMSolverMultistepScheduler", "DPMSolverSDEScheduler"])
-
-try:
-    if not (is_torch_available() and is_transformers_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_torch_and_transformers_objects  # noqa F403
-
-    _import_structure["utils.dummy_torch_and_transformers_objects"] = [
-        name for name in dir(dummy_torch_and_transformers_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["pipelines"].extend(
-        [
-            "AltDiffusionImg2ImgPipeline",
-            "AltDiffusionPipeline",
-            "AmusedImg2ImgPipeline",
-            "AmusedInpaintPipeline",
-            "AmusedPipeline",
-            "AnimateDiffControlNetPipeline",
-            "AnimateDiffPAGPipeline",
-            "AnimateDiffPipeline",
-            "AnimateDiffSDXLPipeline",
-            "AnimateDiffSparseControlNetPipeline",
-            "AnimateDiffVideoToVideoControlNetPipeline",
-            "AnimateDiffVideoToVideoPipeline",
-            "AudioLDM2Pipeline",
-            "AudioLDM2ProjectionModel",
-            "AudioLDM2UNet2DConditionModel",
-            "AudioLDMPipeline",
-            "AuraFlowPipeline",
-            "BlipDiffusionControlNetPipeline",
-            "BlipDiffusionPipeline",
-            "CLIPImageProjection",
-            "CogVideoXPipeline",
-            "CogVideoXVideoToVideoPipeline",
-            "CycleDiffusionPipeline",
-            "FluxControlNetPipeline",
-            "FluxImg2ImgPipeline",
-            "FluxInpaintPipeline",
-            "FluxPipeline",
-            "HunyuanDiTControlNetPipeline",
-            "HunyuanDiTPAGPipeline",
-            "HunyuanDiTPipeline",
-            "I2VGenXLPipeline",
-            "IFImg2ImgPipeline",
-            "IFImg2ImgSuperResolutionPipeline",
-            "IFInpaintingPipeline",
-            "IFInpaintingSuperResolutionPipeline",
-            "IFPipeline",
-            "IFSuperResolutionPipeline",
-            "ImageTextPipelineOutput",
-            "Kandinsky3Img2ImgPipeline",
-            "Kandinsky3Pipeline",
-            "KandinskyCombinedPipeline",
-            "KandinskyImg2ImgCombinedPipeline",
-            "KandinskyImg2ImgPipeline",
-            "KandinskyInpaintCombinedPipeline",
-            "KandinskyInpaintPipeline",
-            "KandinskyPipeline",
-            "KandinskyPriorPipeline",
-            "KandinskyV22CombinedPipeline",
-            "KandinskyV22ControlnetImg2ImgPipeline",
-            "KandinskyV22ControlnetPipeline",
-            "KandinskyV22Img2ImgCombinedPipeline",
-            "KandinskyV22Img2ImgPipeline",
-            "KandinskyV22InpaintCombinedPipeline",
-            "KandinskyV22InpaintPipeline",
-            "KandinskyV22Pipeline",
-            "KandinskyV22PriorEmb2EmbPipeline",
-            "KandinskyV22PriorPipeline",
-            "LatentConsistencyModelImg2ImgPipeline",
-            "LatentConsistencyModelPipeline",
-            "LattePipeline",
-            "LDMTextToImagePipeline",
-            "LEditsPPPipelineStableDiffusion",
-            "LEditsPPPipelineStableDiffusionXL",
-            "LuminaText2ImgPipeline",
-            "MarigoldDepthPipeline",
-            "MarigoldNormalsPipeline",
-            "MusicLDMPipeline",
-            "PaintByExamplePipeline",
-            "PIAPipeline",
-            "PixArtAlphaPipeline",
-            "PixArtSigmaPAGPipeline",
-            "PixArtSigmaPipeline",
-            "SemanticStableDiffusionPipeline",
-            "ShapEImg2ImgPipeline",
-            "ShapEPipeline",
-            "StableAudioPipeline",
-            "StableAudioProjectionModel",
-            "StableCascadeCombinedPipeline",
-            "StableCascadeDecoderPipeline",
-            "StableCascadePriorPipeline",
-            "StableDiffusion3ControlNetInpaintingPipeline",
-            "StableDiffusion3ControlNetPipeline",
-            "StableDiffusion3Img2ImgPipeline",
-            "StableDiffusion3InpaintPipeline",
-            "StableDiffusion3PAGPipeline",
-            "StableDiffusion3Pipeline",
-            "StableDiffusionAdapterPipeline",
-            "StableDiffusionAttendAndExcitePipeline",
-            "StableDiffusionControlNetImg2ImgPipeline",
-            "StableDiffusionControlNetInpaintPipeline",
-            "StableDiffusionControlNetPAGPipeline",
-            "StableDiffusionControlNetPipeline",
-            "StableDiffusionControlNetXSPipeline",
-            "StableDiffusionDepth2ImgPipeline",
-            "StableDiffusionDiffEditPipeline",
-            "StableDiffusionGLIGENPipeline",
-            "StableDiffusionGLIGENTextImagePipeline",
-            "StableDiffusionImageVariationPipeline",
-            "StableDiffusionImg2ImgPipeline",
-            "StableDiffusionInpaintPipeline",
-            "StableDiffusionInpaintPipelineLegacy",
-            "StableDiffusionInstructPix2PixPipeline",
-            "StableDiffusionLatentUpscalePipeline",
-            "StableDiffusionLDM3DPipeline",
-            "StableDiffusionModelEditingPipeline",
-            "StableDiffusionPAGPipeline",
-            "StableDiffusionPanoramaPipeline",
-            "StableDiffusionParadigmsPipeline",
-            "StableDiffusionPipeline",
-            "StableDiffusionPipelineSafe",
-            "StableDiffusionPix2PixZeroPipeline",
-            "StableDiffusionSAGPipeline",
-            "StableDiffusionUpscalePipeline",
-            "StableDiffusionXLAdapterPipeline",
-            "StableDiffusionXLControlNetImg2ImgPipeline",
-            "StableDiffusionXLControlNetInpaintPipeline",
-            "StableDiffusionXLControlNetPAGImg2ImgPipeline",
-            "StableDiffusionXLControlNetPAGPipeline",
-            "StableDiffusionXLControlNetPipeline",
-            "StableDiffusionXLControlNetXSPipeline",
-            "StableDiffusionXLImg2ImgPipeline",
-            "StableDiffusionXLInpaintPipeline",
-            "StableDiffusionXLInstructPix2PixPipeline",
-            "StableDiffusionXLPAGImg2ImgPipeline",
-            "StableDiffusionXLPAGInpaintPipeline",
-            "StableDiffusionXLPAGPipeline",
-            "StableDiffusionXLPipeline",
-            "StableUnCLIPImg2ImgPipeline",
-            "StableUnCLIPPipeline",
-            "StableVideoDiffusionPipeline",
-            "TextToVideoSDPipeline",
-            "TextToVideoZeroPipeline",
-            "TextToVideoZeroSDXLPipeline",
-            "UnCLIPImageVariationPipeline",
-            "UnCLIPPipeline",
-            "UniDiffuserModel",
-            "UniDiffuserPipeline",
-            "UniDiffuserTextDecoder",
-            "VersatileDiffusionDualGuidedPipeline",
-            "VersatileDiffusionImageVariationPipeline",
-            "VersatileDiffusionPipeline",
-            "VersatileDiffusionTextToImagePipeline",
-            "VideoToVideoSDPipeline",
-            "VQDiffusionPipeline",
-            "WuerstchenCombinedPipeline",
-            "WuerstchenDecoderPipeline",
-            "WuerstchenPriorPipeline",
-        ]
-    )
-
-try:
-    if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_torch_and_transformers_and_k_diffusion_objects  # noqa F403
-
-    _import_structure["utils.dummy_torch_and_transformers_and_k_diffusion_objects"] = [
-        name for name in dir(dummy_torch_and_transformers_and_k_diffusion_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["pipelines"].extend(["StableDiffusionKDiffusionPipeline", "StableDiffusionXLKDiffusionPipeline"])
-
-try:
-    if not (is_torch_available() and is_transformers_available() and is_sentencepiece_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_torch_and_transformers_and_sentencepiece_objects  # noqa F403
-
-    _import_structure["utils.dummy_torch_and_transformers_and_sentencepiece_objects"] = [
-        name for name in dir(dummy_torch_and_transformers_and_sentencepiece_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["pipelines"].extend(["KolorsImg2ImgPipeline", "KolorsPAGPipeline", "KolorsPipeline"])
-
-try:
-    if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_torch_and_transformers_and_onnx_objects  # noqa F403
-
-    _import_structure["utils.dummy_torch_and_transformers_and_onnx_objects"] = [
-        name for name in dir(dummy_torch_and_transformers_and_onnx_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["pipelines"].extend(
-        [
-            "OnnxStableDiffusionImg2ImgPipeline",
-            "OnnxStableDiffusionInpaintPipeline",
-            "OnnxStableDiffusionInpaintPipelineLegacy",
-            "OnnxStableDiffusionPipeline",
-            "OnnxStableDiffusionUpscalePipeline",
-            "StableDiffusionOnnxPipeline",
-        ]
-    )
-
-try:
-    if not (is_torch_available() and is_librosa_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_torch_and_librosa_objects  # noqa F403
-
-    _import_structure["utils.dummy_torch_and_librosa_objects"] = [
-        name for name in dir(dummy_torch_and_librosa_objects) if not name.startswith("_")
-    ]
-
-else:
-    _import_structure["pipelines"].extend(["AudioDiffusionPipeline", "Mel"])
-
-try:
-    if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_transformers_and_torch_and_note_seq_objects  # noqa F403
-
-    _import_structure["utils.dummy_transformers_and_torch_and_note_seq_objects"] = [
-        name for name in dir(dummy_transformers_and_torch_and_note_seq_objects) if not name.startswith("_")
-    ]
-
-
-else:
-    _import_structure["pipelines"].extend(["SpectrogramDiffusionPipeline"])
-
-try:
-    if not is_flax_available():
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_flax_objects  # noqa F403
-
-    _import_structure["utils.dummy_flax_objects"] = [
-        name for name in dir(dummy_flax_objects) if not name.startswith("_")
-    ]
-
-
-else:
-    _import_structure["models.controlnet_flax"] = ["FlaxControlNetModel"]
-    _import_structure["models.modeling_flax_utils"] = ["FlaxModelMixin"]
-    _import_structure["models.unets.unet_2d_condition_flax"] = ["FlaxUNet2DConditionModel"]
-    _import_structure["models.vae_flax"] = ["FlaxAutoencoderKL"]
-    _import_structure["pipelines"].extend(["FlaxDiffusionPipeline"])
-    _import_structure["schedulers"].extend(
-        [
-            "FlaxDDIMScheduler",
-            "FlaxDDPMScheduler",
-            "FlaxDPMSolverMultistepScheduler",
-            "FlaxEulerDiscreteScheduler",
-            "FlaxKarrasVeScheduler",
-            "FlaxLMSDiscreteScheduler",
-            "FlaxPNDMScheduler",
-            "FlaxSchedulerMixin",
-            "FlaxScoreSdeVeScheduler",
-        ]
-    )
-
-
-try:
-    if not (is_flax_available() and is_transformers_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_flax_and_transformers_objects  # noqa F403
-
-    _import_structure["utils.dummy_flax_and_transformers_objects"] = [
-        name for name in dir(dummy_flax_and_transformers_objects) if not name.startswith("_")
-    ]
-
-
-else:
-    _import_structure["pipelines"].extend(
-        [
-            "FlaxStableDiffusionControlNetPipeline",
-            "FlaxStableDiffusionImg2ImgPipeline",
-            "FlaxStableDiffusionInpaintPipeline",
-            "FlaxStableDiffusionPipeline",
-            "FlaxStableDiffusionXLPipeline",
-        ]
-    )
-
-try:
-    if not (is_note_seq_available()):
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    from .utils import dummy_note_seq_objects  # noqa F403
-
-    _import_structure["utils.dummy_note_seq_objects"] = [
-        name for name in dir(dummy_note_seq_objects) if not name.startswith("_")
-    ]
-
-
-else:
-    _import_structure["pipelines"].extend(["MidiProcessor"])
-
-if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
-    from .configuration_utils import ConfigMixin
-
-    try:
-        if not is_onnx_available():
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        from .utils.dummy_onnx_objects import *  # noqa F403
-    else:
-        from .pipelines import OnnxRuntimeModel
-
-    try:
-        if not is_torch_available():
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        from .utils.dummy_pt_objects import *  # noqa F403
-    else:
-        from .models import (
-            AsymmetricAutoencoderKL,
-            AuraFlowTransformer2DModel,
-            AutoencoderKL,
-            AutoencoderKLCogVideoX,
-            AutoencoderKLTemporalDecoder,
-            AutoencoderOobleck,
-            AutoencoderTiny,
-            CogVideoXTransformer3DModel,
-            ConsistencyDecoderVAE,
-            ControlNetModel,
-            ControlNetXSAdapter,
-            DiTTransformer2DModel,
-            FluxControlNetModel,
-            FluxMultiControlNetModel,
-            FluxTransformer2DModel,
-            HunyuanDiT2DControlNetModel,
-            HunyuanDiT2DModel,
-            HunyuanDiT2DMultiControlNetModel,
-            I2VGenXLUNet,
-            Kandinsky3UNet,
-            LatteTransformer3DModel,
-            LuminaNextDiT2DModel,
-            ModelMixin,
-            MotionAdapter,
-            MultiAdapter,
-            PixArtTransformer2DModel,
-            PriorTransformer,
-            SD3ControlNetModel,
-            SD3MultiControlNetModel,
-            SD3Transformer2DModel,
-            SparseControlNetModel,
-            StableAudioDiTModel,
-            T2IAdapter,
-            T5FilmDecoder,
-            Transformer2DModel,
-            UNet1DModel,
-            UNet2DConditionModel,
-            UNet2DModel,
-            UNet3DConditionModel,
-            UNetControlNetXSModel,
-            UNetMotionModel,
-            UNetSpatioTemporalConditionModel,
-            UVit2DModel,
-            VQModel,
-        )
-        from .optimization import (
-            get_constant_schedule,
-            get_constant_schedule_with_warmup,
-            get_cosine_schedule_with_warmup,
-            get_cosine_with_hard_restarts_schedule_with_warmup,
-            get_linear_schedule_with_warmup,
-            get_polynomial_decay_schedule_with_warmup,
-            get_scheduler,
-        )
-        from .pipelines import (
-            AudioPipelineOutput,
-            AutoPipelineForImage2Image,
-            AutoPipelineForInpainting,
-            AutoPipelineForText2Image,
-            BlipDiffusionControlNetPipeline,
-            BlipDiffusionPipeline,
-            CLIPImageProjection,
-            ConsistencyModelPipeline,
-            DanceDiffusionPipeline,
-            DDIMPipeline,
-            DDPMPipeline,
-            DiffusionPipeline,
-            DiTPipeline,
-            ImagePipelineOutput,
-            KarrasVePipeline,
-            LDMPipeline,
-            LDMSuperResolutionPipeline,
-            PNDMPipeline,
-            RePaintPipeline,
-            ScoreSdeVePipeline,
-            StableDiffusionMixin,
-        )
-        from .schedulers import (
-            AmusedScheduler,
-            CMStochasticIterativeScheduler,
-            CogVideoXDDIMScheduler,
-            CogVideoXDPMScheduler,
-            DDIMInverseScheduler,
-            DDIMParallelScheduler,
-            DDIMScheduler,
-            DDPMParallelScheduler,
-            DDPMScheduler,
-            DDPMWuerstchenScheduler,
-            DEISMultistepScheduler,
-            DPMSolverMultistepInverseScheduler,
-            DPMSolverMultistepScheduler,
-            DPMSolverSinglestepScheduler,
-            EDMDPMSolverMultistepScheduler,
-            EDMEulerScheduler,
-            EulerAncestralDiscreteScheduler,
-            EulerDiscreteScheduler,
-            FlowMatchEulerDiscreteScheduler,
-            FlowMatchHeunDiscreteScheduler,
-            HeunDiscreteScheduler,
-            IPNDMScheduler,
-            KarrasVeScheduler,
-            KDPM2AncestralDiscreteScheduler,
-            KDPM2DiscreteScheduler,
-            LCMScheduler,
-            PNDMScheduler,
-            RePaintScheduler,
-            SASolverScheduler,
-            SchedulerMixin,
-            ScoreSdeVeScheduler,
-            TCDScheduler,
-            UnCLIPScheduler,
-            UniPCMultistepScheduler,
-            VQDiffusionScheduler,
-        )
-        from .training_utils import EMAModel
-
-    try:
-        if not (is_torch_available() and is_scipy_available()):
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        from .utils.dummy_torch_and_scipy_objects import *  # noqa F403
-    else:
-        from .schedulers import LMSDiscreteScheduler
-
-    try:
-        if not (is_torch_available() and is_torchsde_available()):
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        from .utils.dummy_torch_and_torchsde_objects import *  # noqa F403
-    else:
-        from .schedulers import CosineDPMSolverMultistepScheduler, DPMSolverSDEScheduler
-
-    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 .pipelines import (
-            AltDiffusionImg2ImgPipeline,
-            AltDiffusionPipeline,
-            AmusedImg2ImgPipeline,
-            AmusedInpaintPipeline,
-            AmusedPipeline,
-            AnimateDiffControlNetPipeline,
-            AnimateDiffPAGPipeline,
-            AnimateDiffPipeline,
-            AnimateDiffSDXLPipeline,
-            AnimateDiffSparseControlNetPipeline,
-            AnimateDiffVideoToVideoControlNetPipeline,
-            AnimateDiffVideoToVideoPipeline,
-            AudioLDM2Pipeline,
-            AudioLDM2ProjectionModel,
-            AudioLDM2UNet2DConditionModel,
-            AudioLDMPipeline,
-            AuraFlowPipeline,
-            CLIPImageProjection,
-            CogVideoXPipeline,
-            CogVideoXVideoToVideoPipeline,
-            CycleDiffusionPipeline,
-            FluxControlNetPipeline,
-            FluxImg2ImgPipeline,
-            FluxInpaintPipeline,
-            FluxPipeline,
-            HunyuanDiTControlNetPipeline,
-            HunyuanDiTPAGPipeline,
-            HunyuanDiTPipeline,
-            I2VGenXLPipeline,
-            IFImg2ImgPipeline,
-            IFImg2ImgSuperResolutionPipeline,
-            IFInpaintingPipeline,
-            IFInpaintingSuperResolutionPipeline,
-            IFPipeline,
-            IFSuperResolutionPipeline,
-            ImageTextPipelineOutput,
-            Kandinsky3Img2ImgPipeline,
-            Kandinsky3Pipeline,
-            KandinskyCombinedPipeline,
-            KandinskyImg2ImgCombinedPipeline,
-            KandinskyImg2ImgPipeline,
-            KandinskyInpaintCombinedPipeline,
-            KandinskyInpaintPipeline,
-            KandinskyPipeline,
-            KandinskyPriorPipeline,
-            KandinskyV22CombinedPipeline,
-            KandinskyV22ControlnetImg2ImgPipeline,
-            KandinskyV22ControlnetPipeline,
-            KandinskyV22Img2ImgCombinedPipeline,
-            KandinskyV22Img2ImgPipeline,
-            KandinskyV22InpaintCombinedPipeline,
-            KandinskyV22InpaintPipeline,
-            KandinskyV22Pipeline,
-            KandinskyV22PriorEmb2EmbPipeline,
-            KandinskyV22PriorPipeline,
-            LatentConsistencyModelImg2ImgPipeline,
-            LatentConsistencyModelPipeline,
-            LattePipeline,
-            LDMTextToImagePipeline,
-            LEditsPPPipelineStableDiffusion,
-            LEditsPPPipelineStableDiffusionXL,
-            LuminaText2ImgPipeline,
-            MarigoldDepthPipeline,
-            MarigoldNormalsPipeline,
-            MusicLDMPipeline,
-            PaintByExamplePipeline,
-            PIAPipeline,
-            PixArtAlphaPipeline,
-            PixArtSigmaPAGPipeline,
-            PixArtSigmaPipeline,
-            SemanticStableDiffusionPipeline,
-            ShapEImg2ImgPipeline,
-            ShapEPipeline,
-            StableAudioPipeline,
-            StableAudioProjectionModel,
-            StableCascadeCombinedPipeline,
-            StableCascadeDecoderPipeline,
-            StableCascadePriorPipeline,
-            StableDiffusion3ControlNetPipeline,
-            StableDiffusion3Img2ImgPipeline,
-            StableDiffusion3InpaintPipeline,
-            StableDiffusion3PAGPipeline,
-            StableDiffusion3Pipeline,
-            StableDiffusionAdapterPipeline,
-            StableDiffusionAttendAndExcitePipeline,
-            StableDiffusionControlNetImg2ImgPipeline,
-            StableDiffusionControlNetInpaintPipeline,
-            StableDiffusionControlNetPAGPipeline,
-            StableDiffusionControlNetPipeline,
-            StableDiffusionControlNetXSPipeline,
-            StableDiffusionDepth2ImgPipeline,
-            StableDiffusionDiffEditPipeline,
-            StableDiffusionGLIGENPipeline,
-            StableDiffusionGLIGENTextImagePipeline,
-            StableDiffusionImageVariationPipeline,
-            StableDiffusionImg2ImgPipeline,
-            StableDiffusionInpaintPipeline,
-            StableDiffusionInpaintPipelineLegacy,
-            StableDiffusionInstructPix2PixPipeline,
-            StableDiffusionLatentUpscalePipeline,
-            StableDiffusionLDM3DPipeline,
-            StableDiffusionModelEditingPipeline,
-            StableDiffusionPAGPipeline,
-            StableDiffusionPanoramaPipeline,
-            StableDiffusionParadigmsPipeline,
-            StableDiffusionPipeline,
-            StableDiffusionPipelineSafe,
-            StableDiffusionPix2PixZeroPipeline,
-            StableDiffusionSAGPipeline,
-            StableDiffusionUpscalePipeline,
-            StableDiffusionXLAdapterPipeline,
-            StableDiffusionXLControlNetImg2ImgPipeline,
-            StableDiffusionXLControlNetInpaintPipeline,
-            StableDiffusionXLControlNetPAGImg2ImgPipeline,
-            StableDiffusionXLControlNetPAGPipeline,
-            StableDiffusionXLControlNetPipeline,
-            StableDiffusionXLControlNetXSPipeline,
-            StableDiffusionXLImg2ImgPipeline,
-            StableDiffusionXLInpaintPipeline,
-            StableDiffusionXLInstructPix2PixPipeline,
-            StableDiffusionXLPAGImg2ImgPipeline,
-            StableDiffusionXLPAGInpaintPipeline,
-            StableDiffusionXLPAGPipeline,
-            StableDiffusionXLPipeline,
-            StableUnCLIPImg2ImgPipeline,
-            StableUnCLIPPipeline,
-            StableVideoDiffusionPipeline,
-            TextToVideoSDPipeline,
-            TextToVideoZeroPipeline,
-            TextToVideoZeroSDXLPipeline,
-            UnCLIPImageVariationPipeline,
-            UnCLIPPipeline,
-            UniDiffuserModel,
-            UniDiffuserPipeline,
-            UniDiffuserTextDecoder,
-            VersatileDiffusionDualGuidedPipeline,
-            VersatileDiffusionImageVariationPipeline,
-            VersatileDiffusionPipeline,
-            VersatileDiffusionTextToImagePipeline,
-            VideoToVideoSDPipeline,
-            VQDiffusionPipeline,
-            WuerstchenCombinedPipeline,
-            WuerstchenDecoderPipeline,
-            WuerstchenPriorPipeline,
-        )
-
-    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 .pipelines import StableDiffusionKDiffusionPipeline, StableDiffusionXLKDiffusionPipeline
-
-    try:
-        if not (is_torch_available() and is_transformers_available() and is_sentencepiece_available()):
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        from .utils.dummy_torch_and_transformers_and_sentencepiece_objects import *  # noqa F403
-    else:
-        from .pipelines import KolorsImg2ImgPipeline, KolorsPAGPipeline, KolorsPipeline
-    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 .pipelines import (
-            OnnxStableDiffusionImg2ImgPipeline,
-            OnnxStableDiffusionInpaintPipeline,
-            OnnxStableDiffusionInpaintPipelineLegacy,
-            OnnxStableDiffusionPipeline,
-            OnnxStableDiffusionUpscalePipeline,
-            StableDiffusionOnnxPipeline,
-        )
-
-    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 .pipelines import AudioDiffusionPipeline, Mel
-
-    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 .pipelines import SpectrogramDiffusionPipeline
-
-    try:
-        if not is_flax_available():
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        from .utils.dummy_flax_objects import *  # noqa F403
-    else:
-        from .models.controlnet_flax import FlaxControlNetModel
-        from .models.modeling_flax_utils import FlaxModelMixin
-        from .models.unets.unet_2d_condition_flax import FlaxUNet2DConditionModel
-        from .models.vae_flax import FlaxAutoencoderKL
-        from .pipelines import FlaxDiffusionPipeline
-        from .schedulers import (
-            FlaxDDIMScheduler,
-            FlaxDDPMScheduler,
-            FlaxDPMSolverMultistepScheduler,
-            FlaxEulerDiscreteScheduler,
-            FlaxKarrasVeScheduler,
-            FlaxLMSDiscreteScheduler,
-            FlaxPNDMScheduler,
-            FlaxSchedulerMixin,
-            FlaxScoreSdeVeScheduler,
-        )
-
-    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 .pipelines import (
-            FlaxStableDiffusionControlNetPipeline,
-            FlaxStableDiffusionImg2ImgPipeline,
-            FlaxStableDiffusionInpaintPipeline,
-            FlaxStableDiffusionPipeline,
-            FlaxStableDiffusionXLPipeline,
-        )
-
-    try:
-        if not (is_note_seq_available()):
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        from .utils.dummy_note_seq_objects import *  # noqa F403
-    else:
-        from .pipelines import MidiProcessor
-
-else:
-    import sys
-
-    sys.modules[__name__] = _LazyModule(
-        __name__,
-        globals()["__file__"],
-        _import_structure,
-        module_spec=__spec__,
-        extra_objects={"__version__": __version__},
-    )