Skip to content

Commit 4444379

Browse files
committed
make fix-copies
1 parent 7fefe09 commit 4444379

1 file changed

Lines changed: 15 additions & 0 deletions

File tree

src/diffusers/utils/dummy_pt_objects.py

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

10851085

1086+
class PhotonTransformer2DModel(metaclass=DummyObject):
1087+
_backends = ["torch"]
1088+
1089+
def __init__(self, *args, **kwargs):
1090+
requires_backends(self, ["torch"])
1091+
1092+
@classmethod
1093+
def from_config(cls, *args, **kwargs):
1094+
requires_backends(cls, ["torch"])
1095+
1096+
@classmethod
1097+
def from_pretrained(cls, *args, **kwargs):
1098+
requires_backends(cls, ["torch"])
1099+
1100+
10861101
class PixArtTransformer2DModel(metaclass=DummyObject):
10871102
_backends = ["torch"]
10881103

0 commit comments

Comments
 (0)