Skip to content

Commit a479683

Browse files
committed
make fix-copies
1 parent 020f8ae commit a479683

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

src/diffusers/utils/dummy_torch_and_transformers_objects.py

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

199199

200+
class AniMemoryPipeline(metaclass=DummyObject):
201+
_backends = ["torch", "transformers"]
202+
203+
def __init__(self, *args, **kwargs):
204+
requires_backends(self, ["torch", "transformers"])
205+
206+
@classmethod
207+
def from_config(cls, *args, **kwargs):
208+
requires_backends(cls, ["torch", "transformers"])
209+
210+
@classmethod
211+
def from_pretrained(cls, *args, **kwargs):
212+
requires_backends(cls, ["torch", "transformers"])
213+
214+
200215
class AudioLDM2Pipeline(metaclass=DummyObject):
201216
_backends = ["torch", "transformers"]
202217

0 commit comments

Comments
 (0)