Skip to content

Commit bb07c04

Browse files
committed
make fix-copies
1 parent db2efa3 commit bb07c04

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
@@ -1053,6 +1053,21 @@ def from_pretrained(cls, *args, **kwargs):
10531053
requires_backends(cls, ["torch"])
10541054

10551055

1056+
class ParallelConfig(metaclass=DummyObject):
1057+
_backends = ["torch"]
1058+
1059+
def __init__(self, *args, **kwargs):
1060+
requires_backends(self, ["torch"])
1061+
1062+
@classmethod
1063+
def from_config(cls, *args, **kwargs):
1064+
requires_backends(cls, ["torch"])
1065+
1066+
@classmethod
1067+
def from_pretrained(cls, *args, **kwargs):
1068+
requires_backends(cls, ["torch"])
1069+
1070+
10561071
class PixArtTransformer2DModel(metaclass=DummyObject):
10571072
_backends = ["torch"]
10581073

@@ -1477,6 +1492,10 @@ def attention_backend(*args, **kwargs):
14771492
requires_backends(attention_backend, ["torch"])
14781493

14791494

1495+
def enable_parallelism(*args, **kwargs):
1496+
requires_backends(enable_parallelism, ["torch"])
1497+
1498+
14801499
class ComponentsManager(metaclass=DummyObject):
14811500
_backends = ["torch"]
14821501

0 commit comments

Comments
 (0)