diff --git a/src/diffusers/modular_pipelines/mellon_node_utils.py b/src/diffusers/modular_pipelines/mellon_node_utils.py index 63684fa2fb..aae0427196 100644 --- a/src/diffusers/modular_pipelines/mellon_node_utils.py +++ b/src/diffusers/modular_pipelines/mellon_node_utils.py @@ -180,6 +180,10 @@ class MellonParam: 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 layers(cls, default: int = 4) -> "MellonParam": + return cls(name="layers", label="Layers", type="int", default=default, min=1, max=10, display="slider") + @classmethod def videos(cls) -> "MellonParam": return cls(name="videos", label="Videos", type="video", display="output")