Update ComfyI2I.py

Bug fix for chaining the Mask Ops node
2023-08-29 18:17:46 -04:00
parent b2c225db18
commit 236f6b02ad
1 changed files with 2 additions and 2 deletions
@@ -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",
-}
+}