nodes/LevelPixel-ComfyUI-LevelPixel-Advanced
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change path for Rembg models

Browse Source
This commit is contained in:
Level Pixel Dev
2025-05-29 09:01:54 +06:00
parent ac8536b8ae
commit 53f6914fe6
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/image/birefnet_LP.py
+2 -2
View File
@@ -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"])
nodes/image/rmbg_LP.py
+2 -2
View File
@@ -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"])