Update nodes/SegformerNode.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
2025-03-26 21:44:41 +00:00
parent 877dd93b0c
commit 8fb9b27ea1
1 changed files with 1 additions and 1 deletions
@@ -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: