from typing import TYPE_CHECKING

from ..utils import (

DIFFUSERS_SLOW_IMPORT,

OptionalDependencyNotAvailable,

_LazyModule,

get_objects_from_module,

is_librosa_available,

is_note_seq_available,

is_onnx_available,

is_opencv_available,

is_sentencepiece_available,

is_torch_available,

is_torch_npu_available,

is_transformers_available,

is_transformers_version,

)

# These modules contain pipelines from multiple libraries/frameworks

_dummy_objects = {}

_import_structure = {

"controlnet": [],

"controlnet_hunyuandit": [],

"controlnet_sd3": [],

"deprecated": [],

"latent_diffusion": [],

"ledits_pp": [],

"marigold": [],

"pag": [],

"stable_diffusion": [],

"stable_diffusion_xl": [],

}

try:

if not is_torch_available():

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils import dummy_pt_objects # noqa F403

_dummy_objects.update(get_objects_from_module(dummy_pt_objects))

else:

_import_structure["auto_pipeline"] = [

"AutoPipelineForImage2Image",

"AutoPipelineForInpainting",

"AutoPipelineForText2Image",

"AutoPipelineForText2Audio",

]

_import_structure["consistency_models"] = ["ConsistencyModelPipeline"]

_import_structure["ddim"] = ["DDIMPipeline"]

_import_structure["ddpm"] = ["DDPMPipeline"]

_import_structure["rfd4_proteina"] = ["RFD4ProteinaPipeline"]

_import_structure["dit"] = ["DiTPipeline"]

_import_structure["latent_diffusion"].extend(["LDMSuperResolutionPipeline"])

_import_structure["pipeline_utils"] = [

"AudioPipelineOutput",

"DiffusionPipeline",

"StableDiffusionMixin",

"ImagePipelineOutput",

]

_import_structure["deprecated"].extend(

[

"DanceDiffusionPipeline",

"PNDMPipeline",

"LDMPipeline",

"RePaintPipeline",

"ScoreSdeVePipeline",

"KarrasVePipeline",

]

)

try:

if not (is_torch_available() and is_librosa_available()):

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils import dummy_torch_and_librosa_objects # noqa F403

_dummy_objects.update(get_objects_from_module(dummy_torch_and_librosa_objects))

else:

_import_structure["deprecated"].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

_dummy_objects.update(get_objects_from_module(dummy_transformers_and_torch_and_note_seq_objects))

else:

_import_structure["deprecated"].extend(

[

"MidiProcessor",

"SpectrogramDiffusionPipeline",

]

)

try:

if not (is_torch_available() and is_transformers_available()):

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils import dummy_torch_and_transformers_objects # noqa F403

_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))

else:

_import_structure["deprecated"].extend(

[

"AmusedImg2ImgPipeline",

"AmusedInpaintPipeline",

"AmusedPipeline",

"AudioLDMPipeline",

"BlipDiffusionPipeline",

"I2VGenXLPipeline",

"ImageTextPipelineOutput",

"MusicLDMPipeline",

"PIAPipeline",

"PaintByExamplePipeline",

"SemanticStableDiffusionPipeline",

"StableDiffusionAttendAndExcitePipeline",

"StableDiffusionControlNetXSPipeline",

"StableDiffusionDiffEditPipeline",

"StableDiffusionGLIGENPipeline",

"StableDiffusionGLIGENTextImagePipeline",

"StableDiffusionLDM3DPipeline",

"StableDiffusionPanoramaPipeline",

"StableDiffusionPipelineSafe",

"StableDiffusionSAGPipeline",

"StableDiffusionXLControlNetXSPipeline",

"TextToVideoSDPipeline",

"TextToVideoZeroPipeline",

"TextToVideoZeroSDXLPipeline",

"UnCLIPImageVariationPipeline",

"UnCLIPPipeline",

"UniDiffuserModel",

"UniDiffuserPipeline",

"UniDiffuserTextDecoder",

"VQDiffusionPipeline",

"AltDiffusionPipeline",

"AltDiffusionImg2ImgPipeline",

"CycleDiffusionPipeline",

"StableDiffusionInpaintPipelineLegacy",

"StableDiffusionPix2PixZeroPipeline",

"StableDiffusionParadigmsPipeline",

"StableDiffusionModelEditingPipeline",

"VersatileDiffusionDualGuidedPipeline",

"VersatileDiffusionImageVariationPipeline",

"VersatileDiffusionPipeline",

"VersatileDiffusionTextToImagePipeline",

"VideoToVideoSDPipeline",

"WuerstchenCombinedPipeline",

"WuerstchenDecoderPipeline",

"WuerstchenPriorPipeline",

]

)

_import_structure["ace_step"] = [

"AceStepAudioTokenDetokenizer",

"AceStepAudioTokenizer",

"AceStepConditionEncoder",

"AceStepPipeline",

]

_import_structure["allegro"] = ["AllegroPipeline"]

_import_structure["animatediff"] = [

"AnimateDiffPipeline",

"AnimateDiffControlNetPipeline",

"AnimateDiffSDXLPipeline",

"AnimateDiffSparseControlNetPipeline",

"AnimateDiffVideoToVideoPipeline",

"AnimateDiffVideoToVideoControlNetPipeline",

]

_import_structure["anyflow"] = [

"AnyFlowPipeline",

"AnyFlowFARPipeline",

]

_import_structure["bria"] = ["BriaPipeline"]

_import_structure["bria_fibo"] = ["BriaFiboPipeline", "BriaFiboEditPipeline"]

_import_structure["bria_fibo2"] = ["BriaFibo2Pipeline"]

_import_structure["flux2"] = [

"Flux2Pipeline",

"Flux2KleinPipeline",

"Flux2KleinInpaintPipeline",

"Flux2KleinKVPipeline",

]

_import_structure["flux"] = [

"FluxControlPipeline",

"FluxControlInpaintPipeline",

"FluxControlImg2ImgPipeline",

"FluxControlNetPipeline",

"FluxControlNetImg2ImgPipeline",

"FluxControlNetInpaintPipeline",

"FluxImg2ImgPipeline",

"FluxInpaintPipeline",

"FluxPipeline",

"FluxFillPipeline",

"FluxPriorReduxPipeline",

"ReduxImageEncoder",

"FluxKontextPipeline",

"FluxKontextInpaintPipeline",

]

_import_structure["prx"] = ["PRXPipeline", "PRXPixelPipeline"]

_import_structure["audioldm2"] = [

"AudioLDM2Pipeline",

"AudioLDM2ProjectionModel",

"AudioLDM2UNet2DConditionModel",

]

_import_structure["chroma"] = ["ChromaPipeline", "ChromaImg2ImgPipeline", "ChromaInpaintPipeline"]

_import_structure["cogvideo"] = [

"CogVideoXPipeline",

"CogVideoXImageToVideoPipeline",

"CogVideoXVideoToVideoPipeline",

"CogVideoXFunControlPipeline",

]

_import_structure["cogview3"] = ["CogView3PlusPipeline"]

_import_structure["cogview4"] = ["CogView4Pipeline", "CogView4ControlPipeline"]

_import_structure["consisid"] = ["ConsisIDPipeline"]

_import_structure["cosmos"] = [

"Cosmos2_5_PredictBasePipeline",

"Cosmos2_5_TransferPipeline",

"Cosmos2TextToImagePipeline",

"Cosmos2VideoToWorldPipeline",

"Cosmos3OmniPipeline",

"CosmosActionCondition",

"CosmosTextToWorldPipeline",

"CosmosVideoToWorldPipeline",

]

_import_structure["controlnet"].extend(

[

"BlipDiffusionControlNetPipeline",

"StableDiffusionControlNetImg2ImgPipeline",

"StableDiffusionControlNetInpaintPipeline",

"StableDiffusionControlNetPipeline",

"StableDiffusionXLControlNetImg2ImgPipeline",

"StableDiffusionXLControlNetInpaintPipeline",

"StableDiffusionXLControlNetPipeline",

"StableDiffusionXLControlNetUnionPipeline",

"StableDiffusionXLControlNetUnionInpaintPipeline",

"StableDiffusionXLControlNetUnionImg2ImgPipeline",

]

)

_import_structure["pag"].extend(

[

"StableDiffusionControlNetPAGInpaintPipeline",

"AnimateDiffPAGPipeline",

"KolorsPAGPipeline",

"HunyuanDiTPAGPipeline",

"StableDiffusion3PAGPipeline",

"StableDiffusion3PAGImg2ImgPipeline",

"StableDiffusionPAGPipeline",

"StableDiffusionPAGImg2ImgPipeline",

"StableDiffusionPAGInpaintPipeline",

"StableDiffusionControlNetPAGPipeline",

"StableDiffusionXLPAGPipeline",

"StableDiffusionXLPAGInpaintPipeline",

"StableDiffusionXLControlNetPAGImg2ImgPipeline",

"StableDiffusionXLControlNetPAGPipeline",

"StableDiffusionXLPAGImg2ImgPipeline",

"PixArtSigmaPAGPipeline",

"SanaPAGPipeline",

]

)

_import_structure["controlnet_hunyuandit"].extend(

[

"HunyuanDiTControlNetPipeline",

]

)

_import_structure["controlnet_sd3"].extend(

[

"StableDiffusion3ControlNetPipeline",

"StableDiffusion3ControlNetInpaintingPipeline",

]

)

_import_structure["deepfloyd_if"] = [

"IFImg2ImgPipeline",

"IFImg2ImgSuperResolutionPipeline",

"IFInpaintingPipeline",

"IFInpaintingSuperResolutionPipeline",

"IFPipeline",

"IFSuperResolutionPipeline",

]

_import_structure["diffusion_gemma"] = ["DiffusionGemmaPipeline", "DiffusionGemmaPipelineOutput"]

_import_structure["dreamlite"] = ["DreamLitePipeline", "DreamLiteMobilePipeline", "DreamLitePipelineOutput"]

_import_structure["easyanimate"] = [

"EasyAnimatePipeline",

"EasyAnimateInpaintPipeline",

"EasyAnimateControlPipeline",

]

_import_structure["helios"] = ["HeliosPipeline", "HeliosPyramidPipeline"]

_import_structure["hidream_image"] = ["HiDreamImagePipeline"]

_import_structure["hunyuandit"] = ["HunyuanDiTPipeline"]

_import_structure["hunyuan_video"] = [

"HunyuanVideoPipeline",

"HunyuanSkyreelsImageToVideoPipeline",

"HunyuanVideoImageToVideoPipeline",

"HunyuanVideoFramepackPipeline",

]

_import_structure["hunyuan_video1_5"] = ["HunyuanVideo15Pipeline", "HunyuanVideo15ImageToVideoPipeline"]

_import_structure["hunyuan_image"] = ["HunyuanImagePipeline", "HunyuanImageRefinerPipeline"]

_import_structure["ideogram4"] = ["Ideogram4Pipeline", "Ideogram4PromptEnhancerHead"]

_import_structure["kandinsky"] = [

"KandinskyCombinedPipeline",

"KandinskyImg2ImgCombinedPipeline",

"KandinskyImg2ImgPipeline",

"KandinskyInpaintCombinedPipeline",

"KandinskyInpaintPipeline",

"KandinskyPipeline",

"KandinskyPriorPipeline",

]

_import_structure["kandinsky2_2"] = [

"KandinskyV22CombinedPipeline",

"KandinskyV22ControlnetImg2ImgPipeline",

"KandinskyV22ControlnetPipeline",

"KandinskyV22Img2ImgCombinedPipeline",

"KandinskyV22Img2ImgPipeline",

"KandinskyV22InpaintCombinedPipeline",

"KandinskyV22InpaintPipeline",

"KandinskyV22Pipeline",

"KandinskyV22PriorEmb2EmbPipeline",

"KandinskyV22PriorPipeline",

]

_import_structure["kandinsky3"] = [

"Kandinsky3Img2ImgPipeline",

"Kandinsky3Pipeline",

]

_import_structure["krea2"] = ["Krea2Pipeline"]

_import_structure["latent_consistency_models"] = [

"LatentConsistencyModelImg2ImgPipeline",

"LatentConsistencyModelPipeline",

]

_import_structure["latent_diffusion"].extend(["LDMTextToImagePipeline"])

_import_structure["ledits_pp"].extend(

[

"LEditsPPPipelineStableDiffusion",

"LEditsPPPipelineStableDiffusionXL",

]

)

_import_structure["latte"] = ["LattePipeline"]

_import_structure["llada2"] = ["LLaDA2Pipeline", "LLaDA2PipelineOutput"]

_import_structure["ltx"] = [

"LTXPipeline",

"LTXImageToVideoPipeline",

"LTXConditionPipeline",

"LTXLatentUpsamplePipeline",

"LTXI2VLongMultiPromptPipeline",

]

_import_structure["ltx2"] = [

"LTX2Pipeline",

"LTX2ConditionPipeline",

"LTX2DFRPipeline",

"LTX2DFRPipelineOutput",

"LTX2DFRTemporalRefinePipeline",

"LTX2HDRPipeline",

"LTX2InContextPipeline",

"LTX2ImageToVideoPipeline",

"LTX2LatentUpsamplePipeline",

"LTX2VideoDiffusionDecodePipeline",

]

_import_structure["joyimage"] = [

"JoyImageEditPipeline",

"JoyImageEditPipelineOutput",

"JoyImageEditPlusPipeline",

"JoyImageEditPlusPipelineOutput",

]

_import_structure["lumina"] = ["LuminaPipeline"]

_import_structure["lumina2"] = ["Lumina2Pipeline"]

_import_structure["lucy"] = ["LucyEditPipeline"]

_import_structure["longcat_image"] = ["LongCatImagePipeline", "LongCatImageEditPipeline"]

_import_structure["longcat_audio_dit"] = ["LongCatAudioDiTPipeline"]

_import_structure["marigold"].extend(

[

"MarigoldDepthPipeline",

"MarigoldIntrinsicsPipeline",

"MarigoldNormalsPipeline",

]

)

_import_structure["mochi"] = ["MochiPipeline"]

_import_structure["motif_video"] = [

"MotifVideoPipeline",

"MotifVideoImage2VideoPipeline",

"MotifVideoPipelineOutput",

]

_import_structure["omnigen"] = ["OmniGenPipeline"]

_import_structure["ernie_image"] = ["ErnieImagePipeline"]

_import_structure["ovis_image"] = ["OvisImagePipeline"]

_import_structure["visualcloze"] = ["VisualClozePipeline", "VisualClozeGenerationPipeline"]

_import_structure["pixart_alpha"] = ["PixArtAlphaPipeline", "PixArtSigmaPipeline"]

_import_structure["sana"] = [

"SanaPipeline",

"SanaSprintPipeline",

"SanaControlNetPipeline",

"SanaSprintImg2ImgPipeline",

]

_import_structure["sana_video"] = [

"SanaVideoPipeline",

"SanaImageToVideoPipeline",

]

_import_structure["sana_wm"] = [

"SanaWMPipeline",

"SanaWMLTX2Refiner",

"SanaWMPipelineOutput",

]

_import_structure["shap_e"] = ["ShapEImg2ImgPipeline", "ShapEPipeline"]

_import_structure["stable_audio"] = [

"StableAudioProjectionModel",

"StableAudioPipeline",

]

_import_structure["stable_audio_3"] = [

"StableAudio3AudioToAudioPipeline",

"StableAudio3DurationEmbedder",

"StableAudio3InpaintPipeline",

"StableAudio3Pipeline",

]

_import_structure["stable_cascade"] = [

"StableCascadeCombinedPipeline",

"StableCascadeDecoderPipeline",

"StableCascadePriorPipeline",

]

_import_structure["stable_diffusion"].extend(

[

"CLIPImageProjection",

"StableDiffusionDepth2ImgPipeline",

"StableDiffusionImageVariationPipeline",

"StableDiffusionImg2ImgPipeline",

"StableDiffusionInpaintPipeline",

"StableDiffusionInstructPix2PixPipeline",

"StableDiffusionLatentUpscalePipeline",

"StableDiffusionPipeline",

"StableDiffusionUpscalePipeline",

"StableUnCLIPImg2ImgPipeline",

"StableUnCLIPPipeline",

]

)

_import_structure["aura_flow"] = ["AuraFlowPipeline"]

_import_structure["stable_diffusion_3"] = [

"StableDiffusion3Pipeline",

"StableDiffusion3Img2ImgPipeline",

"StableDiffusion3InpaintPipeline",

]

_import_structure["stable_video_diffusion"] = ["StableVideoDiffusionPipeline"]

_import_structure["stable_diffusion_xl"].extend(

[

"StableDiffusionXLImg2ImgPipeline",

"StableDiffusionXLInpaintPipeline",

"StableDiffusionXLInstructPix2PixPipeline",

"StableDiffusionXLPipeline",

]

)

_import_structure["t2i_adapter"] = [

"StableDiffusionAdapterPipeline",

"StableDiffusionXLAdapterPipeline",

]

_import_structure["wan"] = [

"WanPipeline",

"WanImageToVideoPipeline",

"WanVideoToVideoPipeline",

"WanVACEPipeline",

"WanAnimatePipeline",

]

_import_structure["kandinsky5"] = [

"Kandinsky5T2VPipeline",

"Kandinsky5I2VPipeline",

"Kandinsky5T2IPipeline",

"Kandinsky5I2IPipeline",

]

_import_structure["kandinsky6"] = [

"Kandinsky6SRPipeline",

"Kandinsky6SRPipelineOutput",

"Kandinsky6TI2VAPipeline",

"Kandinsky6TI2VAPipelineOutput",

]

_import_structure["z_image"] = [

"ZImageControlNetInpaintPipeline",

"ZImageControlNetPipeline",

"ZImageImg2ImgPipeline",

"ZImageInpaintPipeline",

"ZImageOmniPipeline",

"ZImagePipeline",

]

_import_structure["skyreels_v2"] = [

"SkyReelsV2DiffusionForcingPipeline",

"SkyReelsV2DiffusionForcingImageToVideoPipeline",

"SkyReelsV2DiffusionForcingVideoToVideoPipeline",

"SkyReelsV2ImageToVideoPipeline",

"SkyReelsV2Pipeline",

]

_import_structure["nucleusmoe_image"] = ["NucleusMoEImagePipeline"]

_import_structure["qwenimage21"] = ["QwenImage21Pipeline"]

_import_structure["qwenimage"] = [

"QwenImagePipeline",

"QwenImageImg2ImgPipeline",

"QwenImageInpaintPipeline",

"QwenImageEditPipeline",

"QwenImageEditPlusPipeline",

"QwenImageEditInpaintPipeline",

"QwenImageControlNetInpaintPipeline",

"QwenImageControlNetPipeline",

"QwenImageLayeredPipeline",

]

_import_structure["chronoedit"] = ["ChronoEditPipeline"]

_import_structure["glm_image"] = ["GlmImagePipeline"]

try:

if not is_onnx_available():

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils import dummy_onnx_objects # noqa F403

_dummy_objects.update(get_objects_from_module(dummy_onnx_objects))

else:

_import_structure["onnx_utils"] = ["OnnxRuntimeModel"]

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

_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_and_onnx_objects))

else:

_import_structure["stable_diffusion"].extend(

[

"OnnxStableDiffusionImg2ImgPipeline",

"OnnxStableDiffusionInpaintPipeline",

"OnnxStableDiffusionPipeline",

"OnnxStableDiffusionUpscalePipeline",

"StableDiffusionOnnxPipeline",

]

)

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,

)

_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_and_sentencepiece_objects))

else:

_import_structure["kolors"] = [

"KolorsPipeline",

"KolorsImg2ImgPipeline",

]

try:

if not (is_torch_available() and is_transformers_available() and is_opencv_available()):

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils import (

dummy_torch_and_transformers_and_opencv_objects,

)

_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_and_opencv_objects))

else:

_import_structure["consisid"] = ["ConsisIDPipeline"]

if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:

try:

if not is_torch_available():

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils.dummy_pt_objects import * # noqa F403

else:

from .auto_pipeline import (

AutoPipelineForImage2Image,

AutoPipelineForInpainting,

AutoPipelineForText2Audio,

AutoPipelineForText2Image,

)

from .consistency_models import ConsistencyModelPipeline

from .ddim import DDIMPipeline

from .ddpm import DDPMPipeline

from .deprecated import (

DanceDiffusionPipeline,

KarrasVePipeline,

LDMPipeline,

PNDMPipeline,

RePaintPipeline,

ScoreSdeVePipeline,

)

from .dit import DiTPipeline

from .latent_diffusion import LDMSuperResolutionPipeline

from .pipeline_utils import (

AudioPipelineOutput,

DiffusionPipeline,

ImagePipelineOutput,

StableDiffusionMixin,

)

from .rfd4_proteina import RFD4ProteinaPipeline

try:

if not (is_torch_available() and is_librosa_available()):

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils.dummy_torch_and_librosa_objects import *

else:

from .deprecated 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 *

else:

from .ace_step import (

AceStepAudioTokenDetokenizer,

AceStepAudioTokenizer,

AceStepConditionEncoder,

AceStepPipeline,

)

from .allegro import AllegroPipeline

from .animatediff import (

AnimateDiffControlNetPipeline,

AnimateDiffPipeline,

AnimateDiffSDXLPipeline,

AnimateDiffSparseControlNetPipeline,

AnimateDiffVideoToVideoControlNetPipeline,

AnimateDiffVideoToVideoPipeline,

)

from .anyflow import (

AnyFlowFARPipeline,

AnyFlowPipeline,

)

from .audioldm2 import (

AudioLDM2Pipeline,

AudioLDM2ProjectionModel,

AudioLDM2UNet2DConditionModel,

)

from .aura_flow import AuraFlowPipeline

from .bria import BriaPipeline

from .bria_fibo import BriaFiboEditPipeline, BriaFiboPipeline

from .bria_fibo2 import BriaFibo2Pipeline

from .chroma import ChromaImg2ImgPipeline, ChromaInpaintPipeline, ChromaPipeline

from .chronoedit import ChronoEditPipeline

from .cogvideo import (

CogVideoXFunControlPipeline,

CogVideoXImageToVideoPipeline,

CogVideoXPipeline,

CogVideoXVideoToVideoPipeline,

)

from .cogview3 import CogView3PlusPipeline

from .cogview4 import CogView4ControlPipeline, CogView4Pipeline

from .controlnet import (

BlipDiffusionControlNetPipeline,

StableDiffusionControlNetImg2ImgPipeline,

StableDiffusionControlNetInpaintPipeline,

StableDiffusionControlNetPipeline,

StableDiffusionXLControlNetImg2ImgPipeline,

StableDiffusionXLControlNetInpaintPipeline,

StableDiffusionXLControlNetPipeline,

StableDiffusionXLControlNetUnionImg2ImgPipeline,

StableDiffusionXLControlNetUnionInpaintPipeline,

StableDiffusionXLControlNetUnionPipeline,

)

from .controlnet_hunyuandit import (

HunyuanDiTControlNetPipeline,

)

from .controlnet_sd3 import StableDiffusion3ControlNetInpaintingPipeline, StableDiffusion3ControlNetPipeline

from .cosmos import (

Cosmos2_5_PredictBasePipeline,

Cosmos2_5_TransferPipeline,

Cosmos2TextToImagePipeline,

Cosmos2VideoToWorldPipeline,

Cosmos3OmniPipeline,

CosmosActionCondition,

CosmosTextToWorldPipeline,

CosmosVideoToWorldPipeline,

)

from .deepfloyd_if import (

IFImg2ImgPipeline,

IFImg2ImgSuperResolutionPipeline,

IFInpaintingPipeline,

IFInpaintingSuperResolutionPipeline,

IFPipeline,

IFSuperResolutionPipeline,

)

from .deprecated import (

AltDiffusionImg2ImgPipeline,

AltDiffusionPipeline,

AmusedImg2ImgPipeline,

AmusedInpaintPipeline,

AmusedPipeline,

AudioLDMPipeline,

BlipDiffusionPipeline,

CycleDiffusionPipeline,

I2VGenXLPipeline,

ImageTextPipelineOutput,

MusicLDMPipeline,

PaintByExamplePipeline,

PIAPipeline,

SemanticStableDiffusionPipeline,

StableDiffusionAttendAndExcitePipeline,

StableDiffusionControlNetXSPipeline,

StableDiffusionDiffEditPipeline,

StableDiffusionGLIGENPipeline,

StableDiffusionGLIGENTextImagePipeline,

StableDiffusionInpaintPipelineLegacy,

StableDiffusionLDM3DPipeline,

StableDiffusionModelEditingPipeline,

StableDiffusionPanoramaPipeline,

StableDiffusionParadigmsPipeline,

StableDiffusionPipelineSafe,

StableDiffusionPix2PixZeroPipeline,

StableDiffusionSAGPipeline,

StableDiffusionXLControlNetXSPipeline,

TextToVideoSDPipeline,

TextToVideoZeroPipeline,

TextToVideoZeroSDXLPipeline,

UnCLIPImageVariationPipeline,

UnCLIPPipeline,

UniDiffuserModel,

UniDiffuserPipeline,

UniDiffuserTextDecoder,

VersatileDiffusionDualGuidedPipeline,

VersatileDiffusionImageVariationPipeline,

VersatileDiffusionPipeline,

VersatileDiffusionTextToImagePipeline,

VideoToVideoSDPipeline,

VQDiffusionPipeline,

WuerstchenCombinedPipeline,

WuerstchenDecoderPipeline,

WuerstchenPriorPipeline,

)

from .diffusion_gemma import DiffusionGemmaPipeline, DiffusionGemmaPipelineOutput

from .dreamlite import (

DreamLiteMobilePipeline,

DreamLitePipeline,

DreamLitePipelineOutput,

)

from .easyanimate import (

EasyAnimateControlPipeline,

EasyAnimateInpaintPipeline,

EasyAnimatePipeline,

)

from .ernie_image import ErnieImagePipeline

from .flux import (

FluxControlImg2ImgPipeline,

FluxControlInpaintPipeline,

FluxControlNetImg2ImgPipeline,

FluxControlNetInpaintPipeline,

FluxControlNetPipeline,

FluxControlPipeline,

FluxFillPipeline,

FluxImg2ImgPipeline,

FluxInpaintPipeline,

FluxKontextInpaintPipeline,

FluxKontextPipeline,

FluxPipeline,

FluxPriorReduxPipeline,

ReduxImageEncoder,

)

from .flux2 import Flux2KleinInpaintPipeline, Flux2KleinKVPipeline, Flux2KleinPipeline, Flux2Pipeline

from .glm_image import GlmImagePipeline

from .helios import HeliosPipeline, HeliosPyramidPipeline

from .hidream_image import HiDreamImagePipeline

from .hunyuan_image import HunyuanImagePipeline, HunyuanImageRefinerPipeline

from .hunyuan_video import (

HunyuanSkyreelsImageToVideoPipeline,

HunyuanVideoFramepackPipeline,

HunyuanVideoImageToVideoPipeline,

HunyuanVideoPipeline,

)

from .hunyuan_video1_5 import HunyuanVideo15ImageToVideoPipeline, HunyuanVideo15Pipeline

from .hunyuandit import HunyuanDiTPipeline

from .ideogram4 import Ideogram4Pipeline, Ideogram4PromptEnhancerHead

from .joyimage import (

JoyImageEditPipeline,

JoyImageEditPipelineOutput,

JoyImageEditPlusPipeline,

JoyImageEditPlusPipelineOutput,

)

from .kandinsky import (

KandinskyCombinedPipeline,

KandinskyImg2ImgCombinedPipeline,

KandinskyImg2ImgPipeline,

KandinskyInpaintCombinedPipeline,

KandinskyInpaintPipeline,

KandinskyPipeline,

KandinskyPriorPipeline,

)

from .kandinsky2_2 import (

KandinskyV22CombinedPipeline,

KandinskyV22ControlnetImg2ImgPipeline,

KandinskyV22ControlnetPipeline,

KandinskyV22Img2ImgCombinedPipeline,

KandinskyV22Img2ImgPipeline,

KandinskyV22InpaintCombinedPipeline,

KandinskyV22InpaintPipeline,

KandinskyV22Pipeline,

KandinskyV22PriorEmb2EmbPipeline,

KandinskyV22PriorPipeline,

)

from .kandinsky3 import (

Kandinsky3Img2ImgPipeline,

Kandinsky3Pipeline,

)

from .kandinsky5 import (

Kandinsky5I2IPipeline,

Kandinsky5I2VPipeline,

Kandinsky5T2IPipeline,

Kandinsky5T2VPipeline,

)

from .kandinsky6 import (

Kandinsky6SRPipeline,

Kandinsky6SRPipelineOutput,

Kandinsky6TI2VAPipeline,

Kandinsky6TI2VAPipelineOutput,

)

from .krea2 import Krea2Pipeline

from .latent_consistency_models import (

LatentConsistencyModelImg2ImgPipeline,

LatentConsistencyModelPipeline,

)

from .latent_diffusion import LDMTextToImagePipeline

from .latte import LattePipeline

from .ledits_pp import (

LEditsPPDiffusionPipelineOutput,

LEditsPPInversionPipelineOutput,

LEditsPPPipelineStableDiffusion,

LEditsPPPipelineStableDiffusionXL,

)

from .llada2 import LLaDA2Pipeline, LLaDA2PipelineOutput

from .longcat_audio_dit import LongCatAudioDiTPipeline

from .longcat_image import LongCatImageEditPipeline, LongCatImagePipeline

from .ltx import (

LTXConditionPipeline,

LTXI2VLongMultiPromptPipeline,

LTXImageToVideoPipeline,

LTXLatentUpsamplePipeline,

LTXPipeline,

)

from .ltx2 import (

LTX2ConditionPipeline,

LTX2DFRPipeline,

LTX2DFRPipelineOutput,

LTX2DFRTemporalRefinePipeline,

LTX2HDRPipeline,

LTX2ImageToVideoPipeline,

LTX2InContextPipeline,

LTX2LatentUpsamplePipeline,

LTX2Pipeline,

LTX2VideoDiffusionDecodePipeline,

)

from .lucy import LucyEditPipeline

from .lumina import LuminaPipeline

from .lumina2 import Lumina2Pipeline

from .marigold import (

MarigoldDepthPipeline,

MarigoldIntrinsicsPipeline,

MarigoldNormalsPipeline,

)

from .mochi import MochiPipeline

from .motif_video import (

MotifVideoImage2VideoPipeline,

MotifVideoPipeline,

MotifVideoPipelineOutput,

)

from .nucleusmoe_image import NucleusMoEImagePipeline

from .omnigen import OmniGenPipeline

from .ovis_image import OvisImagePipeline

from .pag import (

AnimateDiffPAGPipeline,

HunyuanDiTPAGPipeline,

KolorsPAGPipeline,

PixArtSigmaPAGPipeline,

SanaPAGPipeline,

StableDiffusion3PAGImg2ImgPipeline,

StableDiffusion3PAGPipeline,

StableDiffusionControlNetPAGInpaintPipeline,

StableDiffusionControlNetPAGPipeline,

StableDiffusionPAGImg2ImgPipeline,

StableDiffusionPAGInpaintPipeline,

StableDiffusionPAGPipeline,

StableDiffusionXLControlNetPAGImg2ImgPipeline,

StableDiffusionXLControlNetPAGPipeline,

StableDiffusionXLPAGImg2ImgPipeline,

StableDiffusionXLPAGInpaintPipeline,

StableDiffusionXLPAGPipeline,

)

from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline

from .prx import PRXPipeline, PRXPixelPipeline

from .qwenimage import (

QwenImageControlNetInpaintPipeline,

QwenImageControlNetPipeline,

QwenImageEditInpaintPipeline,

QwenImageEditPipeline,

QwenImageEditPlusPipeline,

QwenImageImg2ImgPipeline,

QwenImageInpaintPipeline,

QwenImageLayeredPipeline,

QwenImagePipeline,

)

from .qwenimage21 import QwenImage21Pipeline

from .sana import (

SanaControlNetPipeline,

SanaPipeline,

SanaSprintImg2ImgPipeline,

SanaSprintPipeline,

)

from .sana_video import SanaImageToVideoPipeline, SanaVideoPipeline

from .sana_wm import (

SanaWMLTX2Refiner,

SanaWMPipeline,

SanaWMPipelineOutput,

)

from .shap_e import ShapEImg2ImgPipeline, ShapEPipeline

from .stable_audio import StableAudioPipeline, StableAudioProjectionModel

from .stable_audio_3 import (

StableAudio3AudioToAudioPipeline,

StableAudio3DurationEmbedder,

StableAudio3InpaintPipeline,

StableAudio3Pipeline,

)

from .stable_cascade import (

StableCascadeCombinedPipeline,

StableCascadeDecoderPipeline,

StableCascadePriorPipeline,

)

from .stable_diffusion import (

CLIPImageProjection,

StableDiffusionDepth2ImgPipeline,

StableDiffusionImageVariationPipeline,

StableDiffusionImg2ImgPipeline,

StableDiffusionInpaintPipeline,

StableDiffusionInstructPix2PixPipeline,

StableDiffusionLatentUpscalePipeline,

StableDiffusionPipeline,

StableDiffusionUpscalePipeline,

StableUnCLIPImg2ImgPipeline,

StableUnCLIPPipeline,

)

from .stable_diffusion_3 import (

StableDiffusion3Img2ImgPipeline,

StableDiffusion3InpaintPipeline,

StableDiffusion3Pipeline,

)

from .stable_diffusion_xl import (

StableDiffusionXLImg2ImgPipeline,

StableDiffusionXLInpaintPipeline,

StableDiffusionXLInstructPix2PixPipeline,

StableDiffusionXLPipeline,

)

from .stable_video_diffusion import StableVideoDiffusionPipeline

from .t2i_adapter import (

StableDiffusionAdapterPipeline,

StableDiffusionXLAdapterPipeline,

)

from .visualcloze import VisualClozeGenerationPipeline, VisualClozePipeline

from .wan import (

WanAnimatePipeline,

WanImageToVideoPipeline,

WanPipeline,

WanVACEPipeline,

WanVideoToVideoPipeline,

)

from .z_image import (

ZImageControlNetInpaintPipeline,

ZImageControlNetPipeline,

ZImageImg2ImgPipeline,

ZImageInpaintPipeline,

ZImageOmniPipeline,

ZImagePipeline,

)

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 *

else:

from .stable_diffusion import (

OnnxStableDiffusionImg2ImgPipeline,

OnnxStableDiffusionInpaintPipeline,

OnnxStableDiffusionPipeline,

OnnxStableDiffusionUpscalePipeline,

StableDiffusionOnnxPipeline,

)

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 *

else:

from .kolors import (

KolorsImg2ImgPipeline,

KolorsPipeline,

)

try:

if not (is_torch_available() and is_transformers_available() and is_opencv_available()):

raise OptionalDependencyNotAvailable()

except OptionalDependencyNotAvailable:

from ..utils.dummy_torch_and_transformers_and_opencv_objects import *

else: