Update ImageNode.py

shadowcz007 · Jan 25, 2024 · 3a3da49 · 3a3da49
1 parent 3d68e48
commit 3a3da49
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py
@@ -2120,7 +2120,7 @@ def run(self,width,height,scale_option,image=None,average_color=['on'],fill_colo
                     average_images.append(a_im)
                     hexs.append(hex)
 
-            if mask!=None:
+            try:
                 for mas in mask:
                     for ma in mas:
                         ma=tensor2pil(ma)
@@ -2129,6 +2129,8 @@ def run(self,width,height,scale_option,image=None,average_color=['on'],fill_colo
                         ma=ma.convert('L')
                         ma=pil2tensor(ma)
                         masks.append(ma)
+            except:
+                print('')
 
         return (imgs,average_images,hexs,masks,)