Skip to content

Commit 85bdc94

Browse files
committed
make fix-copies
1 parent 6f3926d commit 85bdc94

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

src/diffusers/utils/dummy_pt_objects.py

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

912912

913+
class TransformerTemporalModel(metaclass=DummyObject):
914+
_backends = ["torch"]
915+
916+
def __init__(self, *args, **kwargs):
917+
requires_backends(self, ["torch"])
918+
919+
@classmethod
920+
def from_config(cls, *args, **kwargs):
921+
requires_backends(cls, ["torch"])
922+
923+
@classmethod
924+
def from_pretrained(cls, *args, **kwargs):
925+
requires_backends(cls, ["torch"])
926+
927+
913928
class UNet1DModel(metaclass=DummyObject):
914929
_backends = ["torch"]
915930

0 commit comments

Comments
 (0)