Skip to content

Commit 15d4fb3

Browse files
author
yiyixuxu
committed
remove dpmsolver and sde-dpmsolver
1 parent c6ec05f commit 15d4fb3

File tree

5 files changed

+858
-96
lines changed

5 files changed

+858
-96
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,7 @@
142142
"DEISMultistepScheduler",
143143
"DPMSolverMultistepInverseScheduler",
144144
"DPMSolverMultistepScheduler",
145+
"DPMSolverMultistepSchedulerLegacy",
145146
"DPMSolverSinglestepScheduler",
146147
"EulerAncestralDiscreteScheduler",
147148
"EulerDiscreteScheduler",
@@ -519,6 +520,7 @@
519520
DEISMultistepScheduler,
520521
DPMSolverMultistepInverseScheduler,
521522
DPMSolverMultistepScheduler,
523+
DPMSolverMultistepSchedulerLegacy,
522524
DPMSolverSinglestepScheduler,
523525
EulerAncestralDiscreteScheduler,
524526
EulerDiscreteScheduler,

src/diffusers/schedulers/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
_dummy_modules.update(get_objects_from_module(dummy_pt_objects))
3939

4040
else:
41-
_import_structure["deprecated"] = ["KarrasVeScheduler", "ScoreSdeVpScheduler"]
41+
_import_structure["deprecated"] = ["DPMSolverMultistepSchedulerLegacy", "KarrasVeScheduler", "ScoreSdeVpScheduler"]
4242
_import_structure["scheduling_amused"] = ["AmusedScheduler"]
4343
_import_structure["scheduling_consistency_decoder"] = ["ConsistencyDecoderScheduler"]
4444
_import_structure["scheduling_consistency_models"] = ["CMStochasticIterativeScheduler"]
@@ -129,7 +129,7 @@
129129
except OptionalDependencyNotAvailable:
130130
from ..utils.dummy_pt_objects import * # noqa F403
131131
else:
132-
from .deprecated import KarrasVeScheduler, ScoreSdeVpScheduler
132+
from .deprecated import DPMSolverMultistepSchedulerLegacy, KarrasVeScheduler, ScoreSdeVpScheduler
133133
from .scheduling_amused import AmusedScheduler
134134
from .scheduling_consistency_decoder import ConsistencyDecoderScheduler
135135
from .scheduling_consistency_models import CMStochasticIterativeScheduler

src/diffusers/schedulers/deprecated/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
_dummy_objects.update(get_objects_from_module(dummy_pt_objects))
2323
else:
24+
_import_structure["scheduling_dpmsolver_multistep_legacy"] = ["DPMSolverMultistepSchedulerLegacy"]
2425
_import_structure["scheduling_karras_ve"] = ["KarrasVeScheduler"]
2526
_import_structure["scheduling_sde_vp"] = ["ScoreSdeVpScheduler"]
2627

@@ -32,6 +33,7 @@
3233
except OptionalDependencyNotAvailable:
3334
from ..utils.dummy_pt_objects import * # noqa F403
3435
else:
36+
from .scheduling_dpmsolver_multistep_legacy import DPMSolverMultistepSchedulerLegacy
3537
from .scheduling_karras_ve import KarrasVeScheduler
3638
from .scheduling_sde_vp import ScoreSdeVpScheduler
3739

0 commit comments

Comments
 (0)