__init__.py 779 Bytes
Newer Older
Patrick von Platen's avatar
Patrick von Platen committed
1
from ..utils import is_inflect_available, is_transformers_available, is_unidecode_available
Patrick von Platen's avatar
Patrick von Platen committed
2
3
4
from .pipeline_bddm import BDDMPipeline
from .pipeline_ddim import DDIMPipeline
from .pipeline_ddpm import DDPMPipeline
5
from .pipeline_latent_diffusion_uncond import LatentDiffusionUncondPipeline
Patrick von Platen's avatar
Patrick von Platen committed
6
from .pipeline_pndm import PNDMPipeline
Patrick von Platen's avatar
Patrick von Platen committed
7
from .pipeline_score_sde_ve import ScoreSdeVePipeline
Patrick von Platen's avatar
Patrick von Platen committed
8
from .pipeline_score_sde_vp import ScoreSdeVpPipeline
Patrick von Platen's avatar
Patrick von Platen committed
9
10


Patrick von Platen's avatar
Patrick von Platen committed
11
# from .pipeline_score_sde import ScoreSdeVePipeline
12
13


Patrick von Platen's avatar
Patrick von Platen committed
14
if is_transformers_available():
Patrick von Platen's avatar
Patrick von Platen committed
15
16
    from .pipeline_glide import GlidePipeline
    from .pipeline_latent_diffusion import LatentDiffusionPipeline
Patrick von Platen's avatar
Patrick von Platen committed
17
18
19


if is_transformers_available() and is_unidecode_available() and is_inflect_available():
Patrick von Platen's avatar
Patrick von Platen committed
20
    from .pipeline_grad_tts import GradTTSPipeline