From 8fb9b27ea1d130caae96a0a7060be2f84623d190 Mon Sep 17 00:00:00 2001 From: Craig Wright Date: Wed, 26 Mar 2025 21:44:41 +0000 Subject: [PATCH] Update nodes/SegformerNode.py Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> --- nodes/SegformerNode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes/SegformerNode.py b/nodes/SegformerNode.py index b11d846..2950e97 100644 --- a/nodes/SegformerNode.py +++ b/nodes/SegformerNode.py @@ -289,7 +289,7 @@ class SegformerNode: import traceback print(f"Error in segmentation: {str(e)}") print(f"Traceback: {traceback.format_exc()}") - return {"result": (image, torch.zeros_like(image[0, :, :, 0]), f"Error: {str(e)}", image), + return {"result": (image, torch.zeros_like(image[0, :, :]), f"Error: {str(e)}", image), "ui": {"images": [image]} if show_on_node else {}} class SegformerNodeMasks: