|
__version__ = "0.30.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_torch_available, |
|
is_torchsde_available, |
|
is_transformers_available, |
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_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 |
|
|
|
_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 |
|
|
|
_import_structure["utils.dummy_pt_objects"] = [name for name in dir(dummy_pt_objects) if not name.startswith("_")] |
|
|
|
else: |
|
_import_structure["models"].extend( |
|
[ |
|
"AsymmetricAutoencoderKL", |
|
"AutoencoderKL", |
|
"AutoencoderKLTemporalDecoder", |
|
"AutoencoderTiny", |
|
"ConsistencyDecoderVAE", |
|
"ControlNetModel", |
|
"ControlNetXSAdapter", |
|
"DiTTransformer2DModel", |
|
"HunyuanDiT2DModel", |
|
"I2VGenXLUNet", |
|
"Kandinsky3UNet", |
|
"ModelMixin", |
|
"MotionAdapter", |
|
"MultiAdapter", |
|
"PixArtTransformer2DModel", |
|
"PriorTransformer", |
|
"SD3ControlNetModel", |
|
"SD3MultiControlNetModel", |
|
"SD3Transformer2DModel", |
|
"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", |
|
"DDIMInverseScheduler", |
|
"DDIMParallelScheduler", |
|
"DDIMScheduler", |
|
"DDPMParallelScheduler", |
|
"DDPMScheduler", |
|
"DDPMWuerstchenScheduler", |
|
"DEISMultistepScheduler", |
|
"DPMSolverMultistepInverseScheduler", |
|
"DPMSolverMultistepScheduler", |
|
"DPMSolverSinglestepScheduler", |
|
"EDMDPMSolverMultistepScheduler", |
|
"EDMEulerScheduler", |
|
"EulerAncestralDiscreteScheduler", |
|
"EulerDiscreteScheduler", |
|
"FlowMatchEulerDiscreteScheduler", |
|
"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 |
|
|
|
_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 |
|
|
|
_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(["DPMSolverSDEScheduler"]) |
|
|
|
try: |
|
if not (is_torch_available() and is_transformers_available()): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
from .utils import dummy_torch_and_transformers_objects |
|
|
|
_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", |
|
"AnimateDiffPipeline", |
|
"AnimateDiffSDXLPipeline", |
|
"AnimateDiffVideoToVideoPipeline", |
|
"AudioLDM2Pipeline", |
|
"AudioLDM2ProjectionModel", |
|
"AudioLDM2UNet2DConditionModel", |
|
"AudioLDMPipeline", |
|
"BlipDiffusionControlNetPipeline", |
|
"BlipDiffusionPipeline", |
|
"CLIPImageProjection", |
|
"CycleDiffusionPipeline", |
|
"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", |
|
"LDMTextToImagePipeline", |
|
"LEditsPPPipelineStableDiffusion", |
|
"LEditsPPPipelineStableDiffusionXL", |
|
"MarigoldDepthPipeline", |
|
"MarigoldNormalsPipeline", |
|
"MusicLDMPipeline", |
|
"PaintByExamplePipeline", |
|
"PIAPipeline", |
|
"PixArtAlphaPipeline", |
|
"PixArtSigmaPipeline", |
|
"SemanticStableDiffusionPipeline", |
|
"ShapEImg2ImgPipeline", |
|
"ShapEPipeline", |
|
"StableCascadeCombinedPipeline", |
|
"StableCascadeDecoderPipeline", |
|
"StableCascadePriorPipeline", |
|
"StableDiffusion3ControlNetPipeline", |
|
"StableDiffusion3Img2ImgPipeline", |
|
"StableDiffusion3Pipeline", |
|
"StableDiffusionAdapterPipeline", |
|
"StableDiffusionAttendAndExcitePipeline", |
|
"StableDiffusionControlNetImg2ImgPipeline", |
|
"StableDiffusionControlNetInpaintPipeline", |
|
"StableDiffusionControlNetPipeline", |
|
"StableDiffusionControlNetXSPipeline", |
|
"StableDiffusionDepth2ImgPipeline", |
|
"StableDiffusionDiffEditPipeline", |
|
"StableDiffusionGLIGENPipeline", |
|
"StableDiffusionGLIGENTextImagePipeline", |
|
"StableDiffusionImageVariationPipeline", |
|
"StableDiffusionImg2ImgPipeline", |
|
"StableDiffusionInpaintPipeline", |
|
"StableDiffusionInpaintPipelineLegacy", |
|
"StableDiffusionInstructPix2PixPipeline", |
|
"StableDiffusionLatentUpscalePipeline", |
|
"StableDiffusionLDM3DPipeline", |
|
"StableDiffusionModelEditingPipeline", |
|
"StableDiffusionPanoramaPipeline", |
|
"StableDiffusionParadigmsPipeline", |
|
"StableDiffusionPipeline", |
|
"StableDiffusionPipelineSafe", |
|
"StableDiffusionPix2PixZeroPipeline", |
|
"StableDiffusionSAGPipeline", |
|
"StableDiffusionUpscalePipeline", |
|
"StableDiffusionXLAdapterPipeline", |
|
"StableDiffusionXLControlNetImg2ImgPipeline", |
|
"StableDiffusionXLControlNetInpaintPipeline", |
|
"StableDiffusionXLControlNetPipeline", |
|
"StableDiffusionXLControlNetXSPipeline", |
|
"StableDiffusionXLImg2ImgPipeline", |
|
"StableDiffusionXLInpaintPipeline", |
|
"StableDiffusionXLInstructPix2PixPipeline", |
|
"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 |
|
|
|
_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_onnx_available()): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
from .utils import dummy_torch_and_transformers_and_onnx_objects |
|
|
|
_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 |
|
|
|
_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 |
|
|
|
_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 |
|
|
|
_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 |
|
|
|
_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 |
|
|
|
_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 * |
|
else: |
|
from .pipelines import OnnxRuntimeModel |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
from .utils.dummy_pt_objects import * |
|
else: |
|
from .models import ( |
|
AsymmetricAutoencoderKL, |
|
AutoencoderKL, |
|
AutoencoderKLTemporalDecoder, |
|
AutoencoderTiny, |
|
ConsistencyDecoderVAE, |
|
ControlNetModel, |
|
ControlNetXSAdapter, |
|
DiTTransformer2DModel, |
|
HunyuanDiT2DModel, |
|
I2VGenXLUNet, |
|
Kandinsky3UNet, |
|
ModelMixin, |
|
MotionAdapter, |
|
MultiAdapter, |
|
PixArtTransformer2DModel, |
|
PriorTransformer, |
|
SD3ControlNetModel, |
|
SD3MultiControlNetModel, |
|
SD3Transformer2DModel, |
|
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, |
|
DDIMInverseScheduler, |
|
DDIMParallelScheduler, |
|
DDIMScheduler, |
|
DDPMParallelScheduler, |
|
DDPMScheduler, |
|
DDPMWuerstchenScheduler, |
|
DEISMultistepScheduler, |
|
DPMSolverMultistepInverseScheduler, |
|
DPMSolverMultistepScheduler, |
|
DPMSolverSinglestepScheduler, |
|
EDMDPMSolverMultistepScheduler, |
|
EDMEulerScheduler, |
|
EulerAncestralDiscreteScheduler, |
|
EulerDiscreteScheduler, |
|
FlowMatchEulerDiscreteScheduler, |
|
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 * |
|
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 * |
|
else: |
|
from .schedulers import DPMSolverSDEScheduler |
|
|
|
try: |
|
if not (is_torch_available() and is_transformers_available()): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
from .utils.dummy_torch_and_transformers_objects import * |
|
else: |
|
from .pipelines import ( |
|
AltDiffusionImg2ImgPipeline, |
|
AltDiffusionPipeline, |
|
AmusedImg2ImgPipeline, |
|
AmusedInpaintPipeline, |
|
AmusedPipeline, |
|
AnimateDiffPipeline, |
|
AnimateDiffSDXLPipeline, |
|
AnimateDiffVideoToVideoPipeline, |
|
AudioLDM2Pipeline, |
|
AudioLDM2ProjectionModel, |
|
AudioLDM2UNet2DConditionModel, |
|
AudioLDMPipeline, |
|
CLIPImageProjection, |
|
CycleDiffusionPipeline, |
|
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, |
|
LDMTextToImagePipeline, |
|
LEditsPPPipelineStableDiffusion, |
|
LEditsPPPipelineStableDiffusionXL, |
|
MarigoldDepthPipeline, |
|
MarigoldNormalsPipeline, |
|
MusicLDMPipeline, |
|
PaintByExamplePipeline, |
|
PIAPipeline, |
|
PixArtAlphaPipeline, |
|
PixArtSigmaPipeline, |
|
SemanticStableDiffusionPipeline, |
|
ShapEImg2ImgPipeline, |
|
ShapEPipeline, |
|
StableCascadeCombinedPipeline, |
|
StableCascadeDecoderPipeline, |
|
StableCascadePriorPipeline, |
|
StableDiffusion3ControlNetPipeline, |
|
StableDiffusion3Img2ImgPipeline, |
|
StableDiffusion3Pipeline, |
|
StableDiffusionAdapterPipeline, |
|
StableDiffusionAttendAndExcitePipeline, |
|
StableDiffusionControlNetImg2ImgPipeline, |
|
StableDiffusionControlNetInpaintPipeline, |
|
StableDiffusionControlNetPipeline, |
|
StableDiffusionControlNetXSPipeline, |
|
StableDiffusionDepth2ImgPipeline, |
|
StableDiffusionDiffEditPipeline, |
|
StableDiffusionGLIGENPipeline, |
|
StableDiffusionGLIGENTextImagePipeline, |
|
StableDiffusionImageVariationPipeline, |
|
StableDiffusionImg2ImgPipeline, |
|
StableDiffusionInpaintPipeline, |
|
StableDiffusionInpaintPipelineLegacy, |
|
StableDiffusionInstructPix2PixPipeline, |
|
StableDiffusionLatentUpscalePipeline, |
|
StableDiffusionLDM3DPipeline, |
|
StableDiffusionModelEditingPipeline, |
|
StableDiffusionPanoramaPipeline, |
|
StableDiffusionParadigmsPipeline, |
|
StableDiffusionPipeline, |
|
StableDiffusionPipelineSafe, |
|
StableDiffusionPix2PixZeroPipeline, |
|
StableDiffusionSAGPipeline, |
|
StableDiffusionUpscalePipeline, |
|
StableDiffusionXLAdapterPipeline, |
|
StableDiffusionXLControlNetImg2ImgPipeline, |
|
StableDiffusionXLControlNetInpaintPipeline, |
|
StableDiffusionXLControlNetPipeline, |
|
StableDiffusionXLControlNetXSPipeline, |
|
StableDiffusionXLImg2ImgPipeline, |
|
StableDiffusionXLInpaintPipeline, |
|
StableDiffusionXLInstructPix2PixPipeline, |
|
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 * |
|
else: |
|
from .pipelines import StableDiffusionKDiffusionPipeline, StableDiffusionXLKDiffusionPipeline |
|
|
|
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 .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 * |
|
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 * |
|
else: |
|
from .pipelines import SpectrogramDiffusionPipeline |
|
|
|
try: |
|
if not is_flax_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
from .utils.dummy_flax_objects import * |
|
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 * |
|
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 * |
|
else: |
|
from .pipelines import MidiProcessor |
|
|
|
else: |
|
import sys |
|
|
|
sys.modules[__name__] = _LazyModule( |
|
__name__, |
|
globals()["__file__"], |
|
_import_structure, |
|
module_spec=__spec__, |
|
extra_objects={"__version__": __version__}, |
|
) |
|
|