Skip to content

Commit f9865ff

Browse files
committed
make fix-copies
1 parent d15d090 commit f9865ff

File tree

2 files changed

+30
-0
lines changed

2 files changed

+30
-0
lines changed

src/diffusers/utils/dummy_pt_objects.py

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

11521152

1153+
class WanVACETransformer3DModel(metaclass=DummyObject):
1154+
_backends = ["torch"]
1155+
1156+
def __init__(self, *args, **kwargs):
1157+
requires_backends(self, ["torch"])
1158+
1159+
@classmethod
1160+
def from_config(cls, *args, **kwargs):
1161+
requires_backends(cls, ["torch"])
1162+
1163+
@classmethod
1164+
def from_pretrained(cls, *args, **kwargs):
1165+
requires_backends(cls, ["torch"])
1166+
1167+
11531168
def get_constant_schedule(*args, **kwargs):
11541169
requires_backends(get_constant_schedule, ["torch"])
11551170

src/diffusers/utils/dummy_torch_and_transformers_objects.py

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

28842884

2885+
class WanVACEPipeline(metaclass=DummyObject):
2886+
_backends = ["torch", "transformers"]
2887+
2888+
def __init__(self, *args, **kwargs):
2889+
requires_backends(self, ["torch", "transformers"])
2890+
2891+
@classmethod
2892+
def from_config(cls, *args, **kwargs):
2893+
requires_backends(cls, ["torch", "transformers"])
2894+
2895+
@classmethod
2896+
def from_pretrained(cls, *args, **kwargs):
2897+
requires_backends(cls, ["torch", "transformers"])
2898+
2899+
28852900
class WanVideoToVideoPipeline(metaclass=DummyObject):
28862901
_backends = ["torch", "transformers"]
28872902

0 commit comments

Comments
 (0)