Spaces:
Paused
Paused
__version__ = "0.27.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, | |
) | |
# 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"], | |
"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", | |
"AutoencoderKL", | |
"AutoencoderKLTemporalDecoder", | |
"AutoencoderTiny", | |
"ConsistencyDecoderVAE", | |
"BrushNetModel", | |
"ControlNetModel", | |
"I2VGenXLUNet", | |
"Kandinsky3UNet", | |
"ModelMixin", | |
"MotionAdapter", | |
"MultiAdapter", | |
"PriorTransformer", | |
"StableCascadeUNet", | |
"T2IAdapter", | |
"T5FilmDecoder", | |
"Transformer2DModel", | |
"UNet1DModel", | |
"UNet2DConditionModel", | |
"UNet2DModel", | |
"UNet3DConditionModel", | |
"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", | |
"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(["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", | |
"AnimateDiffPipeline", | |
"AnimateDiffVideoToVideoPipeline", | |
"AudioLDM2Pipeline", | |
"AudioLDM2ProjectionModel", | |
"AudioLDM2UNet2DConditionModel", | |
"AudioLDMPipeline", | |
"BlipDiffusionControlNetPipeline", | |
"BlipDiffusionPipeline", | |
"CLIPImageProjection", | |
"CycleDiffusionPipeline", | |
"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", | |
"MusicLDMPipeline", | |
"PaintByExamplePipeline", | |
"PIAPipeline", | |
"PixArtAlphaPipeline", | |
"SemanticStableDiffusionPipeline", | |
"ShapEImg2ImgPipeline", | |
"ShapEPipeline", | |
"StableCascadeCombinedPipeline", | |
"StableCascadeDecoderPipeline", | |
"StableCascadePriorPipeline", | |
"StableDiffusionAdapterPipeline", | |
"StableDiffusionAttendAndExcitePipeline", | |
"StableDiffusionBrushNetPipeline", | |
"StableDiffusionBrushNetPowerPaintPipeline", | |
"StableDiffusionControlNetImg2ImgPipeline", | |
"StableDiffusionControlNetInpaintPipeline", | |
"StableDiffusionControlNetPipeline", | |
"StableDiffusionDepth2ImgPipeline", | |
"StableDiffusionDiffEditPipeline", | |
"StableDiffusionGLIGENPipeline", | |
"StableDiffusionGLIGENTextImagePipeline", | |
"StableDiffusionImageVariationPipeline", | |
"StableDiffusionImg2ImgPipeline", | |
"StableDiffusionInpaintPipeline", | |
"StableDiffusionInpaintPipelineLegacy", | |
"StableDiffusionInstructPix2PixPipeline", | |
"StableDiffusionLatentUpscalePipeline", | |
"StableDiffusionLDM3DPipeline", | |
"StableDiffusionModelEditingPipeline", | |
"StableDiffusionPanoramaPipeline", | |
"StableDiffusionParadigmsPipeline", | |
"StableDiffusionPipeline", | |
"StableDiffusionPipelineSafe", | |
"StableDiffusionPix2PixZeroPipeline", | |
"StableDiffusionSAGPipeline", | |
"StableDiffusionUpscalePipeline", | |
"StableDiffusionXLAdapterPipeline", | |
"StableDiffusionXLControlNetImg2ImgPipeline", | |
"StableDiffusionXLControlNetInpaintPipeline", | |
"StableDiffusionXLControlNetPipeline", | |
"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 # 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_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, | |
AutoencoderKL, | |
AutoencoderKLTemporalDecoder, | |
AutoencoderTiny, | |
BrushNetModel, | |
ConsistencyDecoderVAE, | |
ControlNetModel, | |
I2VGenXLUNet, | |
Kandinsky3UNet, | |
ModelMixin, | |
MotionAdapter, | |
MultiAdapter, | |
PriorTransformer, | |
T2IAdapter, | |
T5FilmDecoder, | |
Transformer2DModel, | |
UNet1DModel, | |
UNet2DConditionModel, | |
UNet2DModel, | |
UNet3DConditionModel, | |
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, | |
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 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, | |
AnimateDiffPipeline, | |
AnimateDiffVideoToVideoPipeline, | |
AudioLDM2Pipeline, | |
AudioLDM2ProjectionModel, | |
AudioLDM2UNet2DConditionModel, | |
AudioLDMPipeline, | |
CLIPImageProjection, | |
CycleDiffusionPipeline, | |
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, | |
MusicLDMPipeline, | |
PaintByExamplePipeline, | |
PIAPipeline, | |
PixArtAlphaPipeline, | |
SemanticStableDiffusionPipeline, | |
ShapEImg2ImgPipeline, | |
ShapEPipeline, | |
StableCascadeCombinedPipeline, | |
StableCascadeDecoderPipeline, | |
StableCascadePriorPipeline, | |
StableDiffusionAdapterPipeline, | |
StableDiffusionAttendAndExcitePipeline, | |
StableDiffusionBrushNetPipeline, | |
StableDiffusionBrushNetPowerPaintPipeline, | |
StableDiffusionControlNetImg2ImgPipeline, | |
StableDiffusionControlNetInpaintPipeline, | |
StableDiffusionControlNetPipeline, | |
StableDiffusionDepth2ImgPipeline, | |
StableDiffusionDiffEditPipeline, | |
StableDiffusionGLIGENPipeline, | |
StableDiffusionGLIGENTextImagePipeline, | |
StableDiffusionImageVariationPipeline, | |
StableDiffusionImg2ImgPipeline, | |
StableDiffusionInpaintPipeline, | |
StableDiffusionInpaintPipelineLegacy, | |
StableDiffusionInstructPix2PixPipeline, | |
StableDiffusionLatentUpscalePipeline, | |
StableDiffusionLDM3DPipeline, | |
StableDiffusionModelEditingPipeline, | |
StableDiffusionPanoramaPipeline, | |
StableDiffusionParadigmsPipeline, | |
StableDiffusionPipeline, | |
StableDiffusionPipelineSafe, | |
StableDiffusionPix2PixZeroPipeline, | |
StableDiffusionSAGPipeline, | |
StableDiffusionUpscalePipeline, | |
StableDiffusionXLAdapterPipeline, | |
StableDiffusionXLControlNetImg2ImgPipeline, | |
StableDiffusionXLControlNetInpaintPipeline, | |
StableDiffusionXLControlNetPipeline, | |
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 * # noqa F403 | |
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 * # 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__}, | |
) | |