Skip to content

Commit dab000e

Browse files
authored
[Modular] Video for Mellon (#12924)
num_frames and videos
1 parent 9fb6b89 commit dab000e

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/diffusers/modular_pipelines/mellon_node_utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,14 @@ def num_inference_steps(cls, default: int = 25) -> "MellonParam":
168168
name="num_inference_steps", label="Steps", type="int", default=default, min=1, max=100, display="slider"
169169
)
170170

171+
@classmethod
172+
def num_frames(cls, default: int = 81) -> "MellonParam":
173+
return cls(name="num_frames", label="Frames", type="int", default=default, min=1, max=480, display="slider")
174+
175+
@classmethod
176+
def videos(cls) -> "MellonParam":
177+
return cls(name="videos", label="Videos", type="video", display="output")
178+
171179
@classmethod
172180
def vae(cls) -> "MellonParam":
173181
"""

0 commit comments

Comments
 (0)