Skip to content

Commit 436b772

Browse files
committed
make fix-copies
1 parent 07edfa9 commit 436b772

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

src/diffusers/utils/dummy_torch_and_transformers_objects.py

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

379379

380+
class FasterCacheConfig(metaclass=DummyObject):
381+
_backends = ["torch", "transformers"]
382+
383+
def __init__(self, *args, **kwargs):
384+
requires_backends(self, ["torch", "transformers"])
385+
386+
@classmethod
387+
def from_config(cls, *args, **kwargs):
388+
requires_backends(cls, ["torch", "transformers"])
389+
390+
@classmethod
391+
def from_pretrained(cls, *args, **kwargs):
392+
requires_backends(cls, ["torch", "transformers"])
393+
394+
380395
class FluxControlImg2ImgPipeline(metaclass=DummyObject):
381396
_backends = ["torch", "transformers"]
382397

@@ -2535,3 +2550,7 @@ def from_config(cls, *args, **kwargs):
25352550
@classmethod
25362551
def from_pretrained(cls, *args, **kwargs):
25372552
requires_backends(cls, ["torch", "transformers"])
2553+
2554+
2555+
def apply_faster_cache(*args, **kwargs):
2556+
requires_backends(apply_faster_cache, ["torch", "transformers"])

0 commit comments

Comments
 (0)