__version__ = "0.26.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",

"ControlNetModel",

"Kandinsky3UNet",

"ModelMixin",

"MotionAdapter",

"MultiAdapter",

"PriorTransformer",

"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",

]

)

_import_structure["schedulers"].extend(

[

"AmusedScheduler",

"CMStochasticIterativeScheduler",

"DDIMInverseScheduler",

"DDIMParallelScheduler",

"DDIMScheduler",

"DDPMParallelScheduler",

"DDPMScheduler",

"DDPMWuerstchenScheduler",

"DEISMultistepScheduler",

"DPMSolverMultistepInverseScheduler",

"DPMSolverMultistepScheduler",

"DPMSolverSinglestepScheduler",

"EulerAncestralDiscreteScheduler",

"EulerDiscreteScheduler",

"HeunDiscreteScheduler",

"IPNDMScheduler",

"KarrasVeScheduler",

"KDPM2AncestralDiscreteScheduler",

"KDPM2DiscreteScheduler",

"LCMScheduler",

"PNDMScheduler",

"RePaintScheduler",

"SASolverScheduler",

"SchedulerMixin",

"ScoreSdeVeScheduler",

"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",

"AudioLDM2Pipeline",

"AudioLDM2ProjectionModel",

"AudioLDM2UNet2DConditionModel",

"AudioLDMPipeline",

"BlipDiffusionControlNetPipeline",

"BlipDiffusionPipeline",

"CLIPImageProjection",

"CycleDiffusionPipeline",

"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",

"PixArtAlphaPipeline",

"SemanticStableDiffusionPipeline",

"ShapEImg2ImgPipeline",

"ShapEPipeline",

"StableDiffusionAdapterPipeline",

"StableDiffusionAttendAndExcitePipeline",

"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,

ConsistencyDecoderVAE,

ControlNetModel,

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,

)

from .schedulers import (

AmusedScheduler,

CMStochasticIterativeScheduler,

DDIMInverseScheduler,

DDIMParallelScheduler,

DDIMScheduler,

DDPMParallelScheduler,

DDPMScheduler,

DDPMWuerstchenScheduler,

DEISMultistepScheduler,

DPMSolverMultistepInverseScheduler,

DPMSolverMultistepScheduler,

DPMSolverSinglestepScheduler,

EulerAncestralDiscreteScheduler,

EulerDiscreteScheduler,

HeunDiscreteScheduler,

IPNDMScheduler,

KarrasVeScheduler,

KDPM2AncestralDiscreteScheduler,

KDPM2DiscreteScheduler,

LCMScheduler,

PNDMScheduler,

RePaintScheduler,

SASolverScheduler,

SchedulerMixin,

ScoreSdeVeScheduler,

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,

AudioLDM2Pipeline,

AudioLDM2ProjectionModel,

AudioLDM2UNet2DConditionModel,

AudioLDMPipeline,

CLIPImageProjection,

CycleDiffusionPipeline,

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,

PixArtAlphaPipeline,

SemanticStableDiffusionPipeline,

ShapEImg2ImgPipeline,

ShapEPipeline,

StableDiffusionAdapterPipeline,

StableDiffusionAttendAndExcitePipeline,

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__},

)