Commit d74b804d authored by Patrick von Platen's avatar Patrick von Platen
Browse files

Merge branch 'main' of https://github.com/huggingface/diffusers into main

parents a859b199 22b63d15
...@@ -9,7 +9,15 @@ __version__ = "0.0.4" ...@@ -9,7 +9,15 @@ __version__ = "0.0.4"
from .modeling_utils import ModelMixin from .modeling_utils import ModelMixin
from .models import NCSNpp, TemporalUNet, UNetLDMModel, UNetModel from .models import NCSNpp, TemporalUNet, UNetLDMModel, UNetModel
from .pipeline_utils import DiffusionPipeline from .pipeline_utils import DiffusionPipeline
from .pipelines import BDDMPipeline, DDIMPipeline, DDPMPipeline, PNDMPipeline, ScoreSdeVePipeline, ScoreSdeVpPipeline from .pipelines import (
BDDMPipeline,
DDIMPipeline,
DDPMPipeline,
LatentDiffusionUncondPipeline,
PNDMPipeline,
ScoreSdeVePipeline,
ScoreSdeVpPipeline,
)
from .schedulers import ( from .schedulers import (
DDIMScheduler, DDIMScheduler,
DDPMScheduler, DDPMScheduler,
......
...@@ -2,6 +2,7 @@ from ..utils import is_inflect_available, is_transformers_available, is_unidecod ...@@ -2,6 +2,7 @@ from ..utils import is_inflect_available, is_transformers_available, is_unidecod
from .pipeline_bddm import BDDMPipeline from .pipeline_bddm import BDDMPipeline
from .pipeline_ddim import DDIMPipeline from .pipeline_ddim import DDIMPipeline
from .pipeline_ddpm import DDPMPipeline from .pipeline_ddpm import DDPMPipeline
from .pipeline_latent_diffusion_uncond import LatentDiffusionUncondPipeline
from .pipeline_pndm import PNDMPipeline from .pipeline_pndm import PNDMPipeline
from .pipeline_score_sde_ve import ScoreSdeVePipeline from .pipeline_score_sde_ve import ScoreSdeVePipeline
from .pipeline_score_sde_vp import ScoreSdeVpPipeline from .pipeline_score_sde_vp import ScoreSdeVpPipeline
......
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment