__init__.py 1.82 KB
Newer Older
1
from ..utils import is_flax_available, is_onnx_available, is_torch_available, is_transformers_available
2
3


4
if is_torch_available():
5
    from .dance_diffusion import DanceDiffusionPipeline
6
7
    from .ddim import DDIMPipeline
    from .ddpm import DDPMPipeline
8
    from .latent_diffusion import LDMSuperResolutionPipeline
9
10
    from .latent_diffusion_uncond import LDMPipeline
    from .pndm import PNDMPipeline
Revist's avatar
Revist committed
11
    from .repaint import RePaintPipeline
12
13
14
15
16
    from .score_sde_ve import ScoreSdeVePipeline
    from .stochastic_karras_ve import KarrasVePipeline
else:
    from ..utils.dummy_pt_objects import *  # noqa F403

17
if is_torch_available() and is_transformers_available():
Patrick von Platen's avatar
Patrick von Platen committed
18
    from .alt_diffusion import AltDiffusionImg2ImgPipeline, AltDiffusionPipeline
Patrick von Platen's avatar
Patrick von Platen committed
19
    from .latent_diffusion import LDMTextToImagePipeline
20
    from .stable_diffusion import (
21
        CycleDiffusionPipeline,
22
        StableDiffusionImageVariationPipeline,
23
24
        StableDiffusionImg2ImgPipeline,
        StableDiffusionInpaintPipeline,
25
        StableDiffusionInpaintPipelineLegacy,
26
27
        StableDiffusionPipeline,
    )
28
    from .stable_diffusion_safe import StableDiffusionPipelineSafe
29
30
31
32
33
34
    from .versatile_diffusion import (
        VersatileDiffusionDualGuidedPipeline,
        VersatileDiffusionImageVariationPipeline,
        VersatileDiffusionPipeline,
        VersatileDiffusionTextToImagePipeline,
    )
Will Berman's avatar
Will Berman committed
35
    from .vq_diffusion import VQDiffusionPipeline
36
37

if is_transformers_available() and is_onnx_available():
38
39
40
    from .stable_diffusion import (
        OnnxStableDiffusionImg2ImgPipeline,
        OnnxStableDiffusionInpaintPipeline,
41
        OnnxStableDiffusionInpaintPipelineLegacy,
42
43
44
        OnnxStableDiffusionPipeline,
        StableDiffusionOnnxPipeline,
    )
45
46
47

if is_transformers_available() and is_flax_available():
    from .stable_diffusion import FlaxStableDiffusionPipeline