__init__.py 294 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):
Patrick von Platen's avatar
up  
Patrick von Platen committed
9

anton-l's avatar
anton-l committed
10
11
    class GLIDE:
        pass
Patrick von Platen's avatar
Patrick von Platen committed
12
13


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