Skip to content

Commit 590edd7

Browse files
committed
add
1 parent a8f5134 commit 590edd7

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

src/diffusers/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -860,6 +860,7 @@
860860
EulerDiscreteScheduler,
861861
FlowMatchEulerDiscreteScheduler,
862862
FlowMatchHeunDiscreteScheduler,
863+
FlowMatchLCMScheduler,
863864
HeunDiscreteScheduler,
864865
IPNDMScheduler,
865866
KarrasVeScheduler,

src/diffusers/schedulers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@
6060
_import_structure["scheduling_euler_discrete"] = ["EulerDiscreteScheduler"]
6161
_import_structure["scheduling_flow_match_euler_discrete"] = ["FlowMatchEulerDiscreteScheduler"]
6262
_import_structure["scheduling_flow_match_heun_discrete"] = ["FlowMatchHeunDiscreteScheduler"]
63+
_import_structure["scheduling_flow_match_lcm"] = ["FlowMatchLCMScheduler"]
6364
_import_structure["scheduling_heun_discrete"] = ["HeunDiscreteScheduler"]
6465
_import_structure["scheduling_ipndm"] = ["IPNDMScheduler"]
6566
_import_structure["scheduling_k_dpm_2_ancestral_discrete"] = ["KDPM2AncestralDiscreteScheduler"]
@@ -161,6 +162,7 @@
161162
from .scheduling_euler_discrete import EulerDiscreteScheduler
162163
from .scheduling_flow_match_euler_discrete import FlowMatchEulerDiscreteScheduler
163164
from .scheduling_flow_match_heun_discrete import FlowMatchHeunDiscreteScheduler
165+
from .scheduling_flow_match_lcm import FlowMatchLCMScheduler
164166
from .scheduling_heun_discrete import HeunDiscreteScheduler
165167
from .scheduling_ipndm import IPNDMScheduler
166168
from .scheduling_k_dpm_2_ancestral_discrete import KDPM2AncestralDiscreteScheduler

0 commit comments

Comments
 (0)