From 4b7bf35175e859b0a482d1b3337e870c708760c0 Mon Sep 17 00:00:00 2001
From: d8ahazard <d8ahazard@gmail.com>
Date: Mon, 2 Dec 2024 17:37:24 -0600
Subject: [PATCH] Use local import for rembg
For some reason, sometimes, importing rembg will take like 40s for no good reason. This makes comfy init slow.
This commit fixes that.
---
.../ComfyUIBackend/ExtraNodes/SwarmComfyExtra/SwarmRemBg.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/BuiltinExtensions/ComfyUIBackend/ExtraNodes/SwarmComfyExtra/SwarmRemBg.py b/src/BuiltinExtensions/ComfyUIBackend/ExtraNodes/SwarmComfyExtra/SwarmRemBg.py
index 15a179ec..722519ae 100644
--- a/src/BuiltinExtensions/ComfyUIBackend/ExtraNodes/SwarmComfyExtra/SwarmRemBg.py
+++ b/src/BuiltinExtensions/ComfyUIBackend/ExtraNodes/SwarmComfyExtra/SwarmRemBg.py
@@ -1,7 +1,6 @@
from PIL import Image
import numpy as np
import torch
-from rembg import remove
class SwarmRemBg:
@classmethod
@@ -17,6 +16,7 @@ def INPUT_TYPES(s):
FUNCTION = "rem"
def rem(self, images):
+ from rembg import remove
output = []
masks = []
for image in images: