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: