__init__.py 634 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
2
3
4
5
6
7
8
from .bddm import BDDMPipeline
from .ddim import DDIMPipeline
from .ddpm import DDPMPipeline
from .latent_diffusion_uncond import LatentDiffusionUncondPipeline
from .pndm import PNDMPipeline
from .score_sde_ve import ScoreSdeVePipeline
from .score_sde_vp import ScoreSdeVpPipeline
9
10


Patrick von Platen's avatar
Patrick von Platen committed
11
if is_transformers_available():
12
13
    from .glide import GlidePipeline
    from .latent_diffusion import LatentDiffusionPipeline
Patrick von Platen's avatar
Patrick von Platen committed
14
15
16


if is_transformers_available() and is_unidecode_available() and is_inflect_available():
17
    from .grad_tts import GradTTSPipeline