Spaces:
Runtime error
Runtime error
| __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__}, | |
| ) | |