Skip to content

Commit d8907c7

Browse files
committed
add dummy object for pytorch dpmsolver
1 parent f806068 commit d8907c7

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

src/diffusers/utils/dummy_torch_and_accelerate_objects.py

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

304304

305+
class DPMSolverDiscreteScheduler(metaclass=DummyObject):
306+
_backends = ["torch", "accelerate"]
307+
308+
def __init__(self, *args, **kwargs):
309+
requires_backends(self, ["torch", "accelerate"])
310+
311+
@classmethod
312+
def from_config(cls, *args, **kwargs):
313+
requires_backends(cls, ["torch", "accelerate"])
314+
315+
@classmethod
316+
def from_pretrained(cls, *args, **kwargs):
317+
requires_backends(cls, ["torch", "accelerate"])
318+
319+
305320
class EulerAncestralDiscreteScheduler(metaclass=DummyObject):
306321
_backends = ["torch", "accelerate"]
307322

0 commit comments

Comments
 (0)