__init__.py 649 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
8


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


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


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