diff --git a/src/diffusers/modular_pipelines/mellon_node_utils.py b/src/diffusers/modular_pipelines/mellon_node_utils.py index 4f142a453f..c931055c79 100644 --- a/src/diffusers/modular_pipelines/mellon_node_utils.py +++ b/src/diffusers/modular_pipelines/mellon_node_utils.py @@ -168,6 +168,14 @@ class MellonParam: name="num_inference_steps", label="Steps", type="int", default=default, min=1, max=100, display="slider" ) + @classmethod + def num_frames(cls, default: int = 81) -> "MellonParam": + return cls(name="num_frames", label="Frames", type="int", default=default, min=1, max=480, display="slider") + + @classmethod + def videos(cls) -> "MellonParam": + return cls(name="videos", label="Videos", type="video", display="output") + @classmethod def vae(cls) -> "MellonParam": """