From 236f6b02ad72ee0b092d330c8a35e98f3ce58149 Mon Sep 17 00:00:00 2001
From: ManglerFTW <113069374+ManglerFTW@users.noreply.github.com>
Date: Tue, 29 Aug 2023 18:17:46 -0400
Subject: [PATCH] Update ComfyI2I.py

Bug fix for chaining the Mask Ops node
---
 ComfyI2I.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ComfyI2I.py b/ComfyI2I.py
index 89eaddf..23f8cba 100644
--- a/ComfyI2I.py
+++ b/ComfyI2I.py
@@ -657,7 +657,7 @@ class Mask_Ops:
             inputs = CLIPSegProcessor.from_pretrained(model, cache_dir=cache)
             model = CLIPSegForImageSegmentation.from_pretrained(model, cache_dir=cache)
             
-            image = tensor2pil(image)
+            image = image.convert('RGB')
             cache = os.path.join(MODELS_DIR, 'clipseg')
 
             with torch.no_grad():
@@ -1010,4 +1010,4 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "Mask Ops": "Mask Ops",
     "Inpaint Segments": "Inpaint Segments",
     "Combine and Paste": "Combine and Paste",
-}
\ No newline at end of file
+}