Skip to content

Commit 88084f1

Browse files
committed
make fix-copies
1 parent eae7b9e commit 88084f1

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

938938

939+
class UNet3DModel(metaclass=DummyObject):
940+
_backends = ["torch"]
941+
942+
def __init__(self, *args, **kwargs):
943+
requires_backends(self, ["torch"])
944+
945+
@classmethod
946+
def from_config(cls, *args, **kwargs):
947+
requires_backends(cls, ["torch"])
948+
949+
@classmethod
950+
def from_pretrained(cls, *args, **kwargs):
951+
requires_backends(cls, ["torch"])
952+
953+
939954
class UNetControlNetXSModel(metaclass=DummyObject):
940955
_backends = ["torch"]
941956

0 commit comments

Comments
 (0)