From dc68dfb95eb044fba0db3eec1f736887eac10a29 Mon Sep 17 00:00:00 2001 From: Jordan Phillips Date: Wed, 14 Feb 2024 14:04:40 -0800 Subject: [PATCH] Slightly better empty output handling --- nodes.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/nodes.py b/nodes.py index af4f9e1..baeb378 100644 --- a/nodes.py +++ b/nodes.py @@ -46,12 +46,10 @@ class DeepfaceExtractFacesNode: face_image = comfy_image_from_deepface_image(detected_face["face"]) output_images.append(face_image) - if len(output_images) > 1: - output_image = torch.cat(output_images, dim=0) + if len(output_images) > 0: + return (torch.cat(output_images, dim=0),) else: - output_image = output_images[0] - - return (output_image,) + return ((),) class DeepfaceVerifyNode: def __init__(self): @@ -136,7 +134,10 @@ class DeepfaceVerifyNode: output_images_with_distances.sort(key=lambda row: row[1]) output_images = [row[0] for row in output_images_with_distances] - return (torch.stack(output_images, dim=0),) + if len(output_images) > 0: + return (torch.stack(output_images, dim=0),) + else: + return ((),) NODE_CLASS_MAPPINGS = { "DeepfaceExtractFaces": DeepfaceExtractFacesNode,