__init__.py 293 Bytes
Newer Older
1
from .pipeline_bddm import BDDM
Patrick von Platen's avatar
Patrick von Platen committed
2
3
from .pipeline_ddim import DDIM
from .pipeline_ddpm import DDPM
Patrick von Platen's avatar
Patrick von Platen committed
4
5


anton-l's avatar
anton-l committed
6
7
try:
    from .pipeline_glide import GLIDE
Patrick von Platen's avatar
Patrick von Platen committed
8
except (NameError, ImportError):
anton-l's avatar
anton-l committed
9
10
    class GLIDE:
        pass
Patrick von Platen's avatar
Patrick von Platen committed
11
12


Patrick von Platen's avatar
Patrick von Platen committed
13
from .pipeline_latent_diffusion import LatentDiffusion
14
from .pipeline_pndm import PNDM