Skip to content

Commit 7c54eb1

Browse files
committed
make fix-copies
1 parent 9ee31fb commit 7c54eb1

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

src/diffusers/utils/dummy_torch_and_transformers_objects.py

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

409409

410+
class CosmosVideoToWorldPipeline(metaclass=DummyObject):
411+
_backends = ["torch", "transformers"]
412+
413+
def __init__(self, *args, **kwargs):
414+
requires_backends(self, ["torch", "transformers"])
415+
416+
@classmethod
417+
def from_config(cls, *args, **kwargs):
418+
requires_backends(cls, ["torch", "transformers"])
419+
420+
@classmethod
421+
def from_pretrained(cls, *args, **kwargs):
422+
requires_backends(cls, ["torch", "transformers"])
423+
424+
410425
class CycleDiffusionPipeline(metaclass=DummyObject):
411426
_backends = ["torch", "transformers"]
412427

0 commit comments

Comments
 (0)