From 6b5998ca6a7a5a4283cb93ee1424a3f24fdb93f1 Mon Sep 17 00:00:00 2001
From: aurel-g <github@dev.bzh>
Date: Tue, 5 Nov 2024 00:18:48 +0100
Subject: [PATCH] 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
---
 layered_diffusion.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/layered_diffusion.py b/layered_diffusion.py
index 5ffc79b..ce5fac7 100644
--- 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 = (