Skip to content

Commit 5f494a4

Browse files
committed
update
1 parent 24a743b commit 5f494a4

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

src/diffusers/utils/dummy_torch_and_transformers_objects.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,36 @@ def from_pretrained(cls, *args, **kwargs):
3232
requires_backends(cls, ["torch", "transformers"])
3333

3434

35+
class WanAutoBlocks(metaclass=DummyObject):
36+
_backends = ["torch", "transformers"]
37+
38+
def __init__(self, *args, **kwargs):
39+
requires_backends(self, ["torch", "transformers"])
40+
41+
@classmethod
42+
def from_config(cls, *args, **kwargs):
43+
requires_backends(cls, ["torch", "transformers"])
44+
45+
@classmethod
46+
def from_pretrained(cls, *args, **kwargs):
47+
requires_backends(cls, ["torch", "transformers"])
48+
49+
50+
class WanModularPipeline(metaclass=DummyObject):
51+
_backends = ["torch", "transformers"]
52+
53+
def __init__(self, *args, **kwargs):
54+
requires_backends(self, ["torch", "transformers"])
55+
56+
@classmethod
57+
def from_config(cls, *args, **kwargs):
58+
requires_backends(cls, ["torch", "transformers"])
59+
60+
@classmethod
61+
def from_pretrained(cls, *args, **kwargs):
62+
requires_backends(cls, ["torch", "transformers"])
63+
64+
3565
class AllegroPipeline(metaclass=DummyObject):
3666
_backends = ["torch", "transformers"]
3767

0 commit comments

Comments
 (0)