Skip to content

Commit 0460203

Browse files
make fix-copies
1 parent d4e7e6c commit 0460203

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

898898

899+
class CosmosControlNetModel(metaclass=DummyObject):
900+
_backends = ["torch"]
901+
902+
def __init__(self, *args, **kwargs):
903+
requires_backends(self, ["torch"])
904+
905+
@classmethod
906+
def from_config(cls, *args, **kwargs):
907+
requires_backends(cls, ["torch"])
908+
909+
@classmethod
910+
def from_pretrained(cls, *args, **kwargs):
911+
requires_backends(cls, ["torch"])
912+
913+
899914
class CosmosTransformer3DModel(metaclass=DummyObject):
900915
_backends = ["torch"]
901916

0 commit comments

Comments
 (0)