Skip to content

Commit 1cb73cb

Browse files
authored
import for FlowMatchLCMScheduler (huggingface#11318)
* add * fix-copies
1 parent ba6008a commit 1cb73cb

File tree

3 files changed

+18
-0
lines changed

3 files changed

+18
-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

src/diffusers/utils/dummy_pt_objects.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1763,6 +1763,21 @@ def from_pretrained(cls, *args, **kwargs):
17631763
requires_backends(cls, ["torch"])
17641764

17651765

1766+
class FlowMatchLCMScheduler(metaclass=DummyObject):
1767+
_backends = ["torch"]
1768+
1769+
def __init__(self, *args, **kwargs):
1770+
requires_backends(self, ["torch"])
1771+
1772+
@classmethod
1773+
def from_config(cls, *args, **kwargs):
1774+
requires_backends(cls, ["torch"])
1775+
1776+
@classmethod
1777+
def from_pretrained(cls, *args, **kwargs):
1778+
requires_backends(cls, ["torch"])
1779+
1780+
17661781
class HeunDiscreteScheduler(metaclass=DummyObject):
17671782
_backends = ["torch"]
17681783

0 commit comments

Comments
 (0)