__init__.py 4.66 KB
Newer Older
1
from ..utils import (
2
    OptionalDependencyNotAvailable,
3
    is_flax_available,
4
    is_k_diffusion_available,
5
6
7
8
9
    is_librosa_available,
    is_onnx_available,
    is_torch_available,
    is_transformers_available,
)
10
11


12
13
14
15
16
17
try:
    if not is_torch_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from ..utils.dummy_pt_objects import *  # noqa F403
else:
18
    from .dance_diffusion import DanceDiffusionPipeline
19
20
    from .ddim import DDIMPipeline
    from .ddpm import DDPMPipeline
Kashif Rasul's avatar
Kashif Rasul committed
21
    from .dit import DiTPipeline
22
    from .latent_diffusion import LDMSuperResolutionPipeline
23
    from .latent_diffusion_uncond import LDMPipeline
24
    from .pipeline_utils import AudioPipelineOutput, DiffusionPipeline, ImagePipelineOutput
25
    from .pndm import PNDMPipeline
Revist's avatar
Revist committed
26
    from .repaint import RePaintPipeline
27
28
29
    from .score_sde_ve import ScoreSdeVePipeline
    from .stochastic_karras_ve import KarrasVePipeline

30
31
32
33
34
try:
    if not (is_torch_available() and is_librosa_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from ..utils.dummy_torch_and_librosa_objects import *  # noqa F403
35
else:
36
    from .audio_diffusion import AudioDiffusionPipeline, Mel
37

38
39
40
41
42
43
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:
Patrick von Platen's avatar
Patrick von Platen committed
44
    from .alt_diffusion import AltDiffusionImg2ImgPipeline, AltDiffusionPipeline
Patrick von Platen's avatar
Patrick von Platen committed
45
    from .latent_diffusion import LDMTextToImagePipeline
46
    from .paint_by_example import PaintByExamplePipeline
47
    from .semantic_stable_diffusion import SemanticStableDiffusionPipeline
48
    from .stable_diffusion import (
49
        CycleDiffusionPipeline,
YiYi Xu's avatar
YiYi Xu committed
50
        StableDiffusionAttendAndExcitePipeline,
51
        StableDiffusionControlNetPipeline,
52
        StableDiffusionDepth2ImgPipeline,
53
        StableDiffusionImageVariationPipeline,
54
55
        StableDiffusionImg2ImgPipeline,
        StableDiffusionInpaintPipeline,
56
        StableDiffusionInpaintPipelineLegacy,
57
        StableDiffusionInstructPix2PixPipeline,
58
        StableDiffusionLatentUpscalePipeline,
59
        StableDiffusionPanoramaPipeline,
60
        StableDiffusionPipeline,
61
        StableDiffusionPix2PixZeroPipeline,
62
        StableDiffusionSAGPipeline,
63
        StableDiffusionUpscalePipeline,
Will Berman's avatar
Will Berman committed
64
65
        StableUnCLIPImg2ImgPipeline,
        StableUnCLIPPipeline,
66
    )
67
    from .stable_diffusion_safe import StableDiffusionPipelineSafe
68
    from .text_to_video_synthesis import TextToVideoSDPipeline
Will Berman's avatar
Will Berman committed
69
    from .unclip import UnCLIPImageVariationPipeline, UnCLIPPipeline
70
71
72
73
74
75
    from .versatile_diffusion import (
        VersatileDiffusionDualGuidedPipeline,
        VersatileDiffusionImageVariationPipeline,
        VersatileDiffusionPipeline,
        VersatileDiffusionTextToImagePipeline,
    )
Will Berman's avatar
Will Berman committed
76
    from .vq_diffusion import VQDiffusionPipeline
77

78
79
80
81
82
83
84
85
try:
    if not is_onnx_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from ..utils.dummy_onnx_objects import *  # noqa F403
else:
    from .onnx_utils import OnnxRuntimeModel

86
87
88
89
90
91
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:
92
93
94
    from .stable_diffusion import (
        OnnxStableDiffusionImg2ImgPipeline,
        OnnxStableDiffusionInpaintPipeline,
95
        OnnxStableDiffusionInpaintPipelineLegacy,
96
        OnnxStableDiffusionPipeline,
97
        OnnxStableDiffusionUpscalePipeline,
98
99
        StableDiffusionOnnxPipeline,
    )
100

101
102
103
104
105
106
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:
107
108
    from .stable_diffusion import StableDiffusionKDiffusionPipeline

109
110
111
112
113
114
115
116
try:
    if not is_flax_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from ..utils.dummy_flax_objects import *  # noqa F403
else:
    from .pipeline_flax_utils import FlaxDiffusionPipeline

117
118
119
120
121
122
123

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:
124
125
126
127
128
    from .stable_diffusion import (
        FlaxStableDiffusionImg2ImgPipeline,
        FlaxStableDiffusionInpaintPipeline,
        FlaxStableDiffusionPipeline,
    )