Skip to content

Commit e110262

Browse files
committed
add dpm-solver pytorch in stable-diffusion pipeline
1 parent ee5cc97 commit e110262

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
EulerDiscreteScheduler,
1616
LMSDiscreteScheduler,
1717
PNDMScheduler,
18+
DPMSolverDiscreteScheduler,
1819
)
1920
from ...utils import deprecate, logging
2021
from . import StableDiffusionPipelineOutput
@@ -59,7 +60,8 @@ def __init__(
5960
tokenizer: CLIPTokenizer,
6061
unet: UNet2DConditionModel,
6162
scheduler: Union[
62-
DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler
63+
DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler,
64+
DPMSolverDiscreteScheduler,
6365
],
6466
safety_checker: StableDiffusionSafetyChecker,
6567
feature_extractor: CLIPFeatureExtractor,

0 commit comments

Comments
 (0)