Skip to content

Commit 4f52e34

Browse files
committed
make fix-copies
1 parent 1e7217f commit 4f52e34

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

src/diffusers/utils/dummy_pt_objects.py

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

260260

261+
class AttentionBackendName(metaclass=DummyObject):
262+
_backends = ["torch"]
263+
264+
def __init__(self, *args, **kwargs):
265+
requires_backends(self, ["torch"])
266+
267+
@classmethod
268+
def from_config(cls, *args, **kwargs):
269+
requires_backends(cls, ["torch"])
270+
271+
@classmethod
272+
def from_pretrained(cls, *args, **kwargs):
273+
requires_backends(cls, ["torch"])
274+
275+
261276
class AuraFlowTransformer2DModel(metaclass=DummyObject):
262277
_backends = ["torch"]
263278

@@ -1353,6 +1368,10 @@ def from_pretrained(cls, *args, **kwargs):
13531368
requires_backends(cls, ["torch"])
13541369

13551370

1371+
def attention_backend(*args, **kwargs):
1372+
requires_backends(attention_backend, ["torch"])
1373+
1374+
13561375
class ComponentsManager(metaclass=DummyObject):
13571376
_backends = ["torch"]
13581377

0 commit comments

Comments
 (0)