Skip to content

Commit f57e7bd

Browse files
authored
Add EDMEulerScheduler (#7109)
* Add EDMEulerScheduler * address review comments * fix import * fix test * add tests * add co-author Co-authored-by: @dg845 [email protected]
1 parent 3e3d469 commit f57e7bd

File tree

8 files changed

+616
-0
lines changed

8 files changed

+616
-0
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@
144144
"DPMSolverMultistepInverseScheduler",
145145
"DPMSolverMultistepScheduler",
146146
"DPMSolverSinglestepScheduler",
147+
"EDMEulerScheduler",
147148
"EulerAncestralDiscreteScheduler",
148149
"EulerDiscreteScheduler",
149150
"HeunDiscreteScheduler",
@@ -526,6 +527,7 @@
526527
DPMSolverMultistepInverseScheduler,
527528
DPMSolverMultistepScheduler,
528529
DPMSolverSinglestepScheduler,
530+
EDMEulerScheduler,
529531
EulerAncestralDiscreteScheduler,
530532
EulerDiscreteScheduler,
531533
HeunDiscreteScheduler,

src/diffusers/schedulers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@
5252
_import_structure["scheduling_dpmsolver_multistep"] = ["DPMSolverMultistepScheduler"]
5353
_import_structure["scheduling_dpmsolver_multistep_inverse"] = ["DPMSolverMultistepInverseScheduler"]
5454
_import_structure["scheduling_dpmsolver_singlestep"] = ["DPMSolverSinglestepScheduler"]
55+
_import_structure["scheduling_edm_euler"] = ["EDMEulerScheduler"]
5556
_import_structure["scheduling_euler_ancestral_discrete"] = ["EulerAncestralDiscreteScheduler"]
5657
_import_structure["scheduling_euler_discrete"] = ["EulerDiscreteScheduler"]
5758
_import_structure["scheduling_heun_discrete"] = ["HeunDiscreteScheduler"]
@@ -144,6 +145,7 @@
144145
from .scheduling_dpmsolver_multistep import DPMSolverMultistepScheduler
145146
from .scheduling_dpmsolver_multistep_inverse import DPMSolverMultistepInverseScheduler
146147
from .scheduling_dpmsolver_singlestep import DPMSolverSinglestepScheduler
148+
from .scheduling_edm_euler import EDMEulerScheduler
147149
from .scheduling_euler_ancestral_discrete import EulerAncestralDiscreteScheduler
148150
from .scheduling_euler_discrete import EulerDiscreteScheduler
149151
from .scheduling_heun_discrete import HeunDiscreteScheduler

0 commit comments

Comments
 (0)