Skip to content

Commit 80ccc90

Browse files
committed
make fix-copies
1 parent 6601fdc commit 80ccc90

File tree

4 files changed

+21
-2
lines changed

4 files changed

+21
-2
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,7 @@
315315
"MarigoldDepthPipeline",
316316
"MarigoldNormalsPipeline",
317317
"MochiPipeline",
318+
"MochiVideoToVideoPipeline",
318319
"MusicLDMPipeline",
319320
"PaintByExamplePipeline",
320321
"PIAPipeline",
@@ -780,6 +781,7 @@
780781
MarigoldDepthPipeline,
781782
MarigoldNormalsPipeline,
782783
MochiPipeline,
784+
MochiVideoToVideoPipeline,
783785
MusicLDMPipeline,
784786
PaintByExamplePipeline,
785787
PIAPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@
247247
"MarigoldNormalsPipeline",
248248
]
249249
)
250-
_import_structure["mochi"] = ["MochiPipeline"]
250+
_import_structure["mochi"] = ["MochiPipeline", "MochiVideoToVideoPipeline"]
251251
_import_structure["musicldm"] = ["MusicLDMPipeline"]
252252
_import_structure["paint_by_example"] = ["PaintByExamplePipeline"]
253253
_import_structure["pia"] = ["PIAPipeline"]
@@ -572,7 +572,7 @@
572572
MarigoldDepthPipeline,
573573
MarigoldNormalsPipeline,
574574
)
575-
from .mochi import MochiPipeline
575+
from .mochi import MochiPipeline, MochiVideoToVideoPipeline
576576
from .musicldm import MusicLDMPipeline
577577
from .pag import (
578578
AnimateDiffPAGPipeline,

src/diffusers/pipelines/mochi/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
2424
else:
2525
_import_structure["pipeline_mochi"] = ["MochiPipeline"]
26+
_import_structure["pipeline_mochi_video2video"] = ["MochiVideoToVideoPipeline"]
2627

2728
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
2829
try:
@@ -33,6 +34,7 @@
3334
from ...utils.dummy_torch_and_transformers_objects import *
3435
else:
3536
from .pipeline_mochi import MochiPipeline
37+
from .pipeline_mochi_video2video import MochiVideoToVideoPipeline
3638

3739
else:
3840
import sys

src/diffusers/utils/dummy_torch_and_transformers_objects.py

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

10691069

1070+
class MochiVideoToVideoPipeline(metaclass=DummyObject):
1071+
_backends = ["torch", "transformers"]
1072+
1073+
def __init__(self, *args, **kwargs):
1074+
requires_backends(self, ["torch", "transformers"])
1075+
1076+
@classmethod
1077+
def from_config(cls, *args, **kwargs):
1078+
requires_backends(cls, ["torch", "transformers"])
1079+
1080+
@classmethod
1081+
def from_pretrained(cls, *args, **kwargs):
1082+
requires_backends(cls, ["torch", "transformers"])
1083+
1084+
10701085
class MusicLDMPipeline(metaclass=DummyObject):
10711086
_backends = ["torch", "transformers"]
10721087

0 commit comments

Comments
 (0)