Adding explicit names for outputs of some nodes

Added explicit names for outputs of some nodes that have multiple outputs of the same type and for which there could be ambiguity
huchenlei · Nov 4, 2024 · 6b5998c · 6b5998c
1 parent 6e4aeb2
commit 6b5998c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/layered_diffusion.py b/layered_diffusion.py
@@ -164,6 +164,7 @@ def INPUT_TYPES(s):
 
     MAX_FRAMES = 3
     RETURN_TYPES = ("IMAGE",) * MAX_FRAMES
+    RETURN_NAMES = tuple(f"image{i + 1}" for i in range(MAX_FRAMES))
 
     def decode(
         self,
@@ -441,6 +442,7 @@ def INPUT_TYPES(s):
         }
 
     RETURN_TYPES = ("MODEL", "CONDITIONING", "CONDITIONING")
+    RETURN_NAMES = ("MODEL", "positive", "negative")
     FUNCTION = "apply_layered_diffusion"
     CATEGORY = "layer_diffuse"
     MODELS = (
@@ -566,6 +568,7 @@ def INPUT_TYPES(s):
         }
 
     RETURN_TYPES = ("MODEL", "CONDITIONING", "CONDITIONING")
+    RETURN_NAMES = ("MODEL", "positive", "negative")
     FUNCTION = "apply_layered_diffusion"
     CATEGORY = "layer_diffuse"
     MODELS = (