From afefeca48f749644bb15d0fa1094ff7704207dd6 Mon Sep 17 00:00:00 2001
From: Acly <aclysia@gmail.com>
Date: Mon, 3 Jun 2024 23:16:59 +0200
Subject: [PATCH] Add a batch dimension to masks loaded by LoadMaskBase64

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 2e91e5b..bf03f45 100644
--- a/nodes.py
+++ b/nodes.py
@@ -49,7 +49,7 @@ class LoadMaskBase64:
         img = torch.from_numpy(img)
         if img.dim() == 3:  # RGB(A) input, use red channel
             img = img[:, :, 0]
-        return (img,)
+        return (img.unsqueeze(0),)
 
 
 class SendImageWebSocket: