__init__.py 703 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
5
from .pipeline_bddm import BDDMPipeline
from .pipeline_ddim import DDIMPipeline
from .pipeline_ddpm import DDPMPipeline
from .pipeline_pndm import PNDMPipeline
Patrick von Platen's avatar
Patrick von Platen committed
6
from .pipeline_score_sde_ve import ScoreSdeVePipeline
Patrick von Platen's avatar
Patrick von Platen committed
7
from .pipeline_score_sde_vp import ScoreSdeVpPipeline
Patrick von Platen's avatar
Patrick von Platen committed
8
9


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


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


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