From 53f6914fe63a16c4dcb0543ee97ad6d729279996 Mon Sep 17 00:00:00 2001 From: Level Pixel Dev Date: Thu, 29 May 2025 09:01:54 +0600 Subject: [PATCH] Change path for Rembg models --- nodes/image/birefnet_LP.py | 4 ++-- nodes/image/rmbg_LP.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/nodes/image/birefnet_LP.py b/nodes/image/birefnet_LP.py index 142d343..4d6d781 100644 --- a/nodes/image/birefnet_LP.py +++ b/nodes/image/birefnet_LP.py @@ -12,7 +12,7 @@ import cv2 device = "cuda" if torch.cuda.is_available() else "cpu" -folder_paths.add_model_folder_path("rmbg", os.path.join(folder_paths.models_dir, "RMBG")) +folder_paths.add_model_folder_path("rembg", os.path.join(folder_paths.models_dir, "rembg")) MODEL_CONFIG = { "BiRefNet-HR-matting": { @@ -181,7 +181,7 @@ class BiRefNetModel: def __init__(self): self.model = None self.current_model_version = None - self.base_cache_dir = os.path.join(folder_paths.models_dir, "RMBG") + self.base_cache_dir = os.path.join(folder_paths.models_dir, "rembg") def get_cache_dir(self, model_name): return os.path.join(self.base_cache_dir, MODEL_CONFIG[model_name]["cache_dir"]) diff --git a/nodes/image/rmbg_LP.py b/nodes/image/rmbg_LP.py index 48f6e0d..a404118 100644 --- a/nodes/image/rmbg_LP.py +++ b/nodes/image/rmbg_LP.py @@ -15,7 +15,7 @@ import types device = "cuda" if torch.cuda.is_available() else "cpu" -folder_paths.add_model_folder_path("rmbg", os.path.join(folder_paths.models_dir, "RMBG")) +folder_paths.add_model_folder_path("rembg", os.path.join(folder_paths.models_dir, "rembg")) AVAILABLE_MODELS = { "RMBG-2.0": { @@ -71,7 +71,7 @@ class BaseModelLoader: def __init__(self): self.model = None self.current_model_version = None - self.base_cache_dir = os.path.join(folder_paths.models_dir, "RMBG") + self.base_cache_dir = os.path.join(folder_paths.models_dir, "rembg") def get_cache_dir(self, model_name): cache_path = os.path.join(self.base_cache_dir, AVAILABLE_MODELS[model_name]["cache_dir"])