From 4d514aae3afdcd12224ef0075dade51dd1b137a8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=81lvaro=20Somoza?= Date: Wed, 7 Jan 2026 19:07:38 -0300 Subject: [PATCH] num_frames and videos --- src/diffusers/modular_pipelines/mellon_node_utils.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/diffusers/modular_pipelines/mellon_node_utils.py b/src/diffusers/modular_pipelines/mellon_node_utils.py index 4f142a453f3b..c931055c794f 100644 --- a/src/diffusers/modular_pipelines/mellon_node_utils.py +++ b/src/diffusers/modular_pipelines/mellon_node_utils.py @@ -168,6 +168,14 @@ def num_inference_steps(cls, default: int = 25) -> "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": """