From f5ee452e8c1d01cc95ceedc28ba6ce583479d507 Mon Sep 17 00:00:00 2001
From: moon7star9 <yueqixingjiu@qq.com>
Date: Fri, 21 Feb 2025 16:37:49 +0800
Subject: [PATCH] modify models_path

---
 nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index b9c3265..7802496 100644
--- a/nodes.py
+++ b/nodes.py
@@ -6,11 +6,11 @@ from PIL import Image
 import torch.nn.functional as F  
 import os  
 import cv2  
+import folder_paths
 
 torch.set_float32_matmul_precision(["high", "highest"][0])  
 
-current_path = os.getcwd()  
-models_path = os.path.join(current_path, "models", "BiRefNet")  
+models_path = os.path.join(folder_paths.models_dir, "BiRefNet")
 
 # 模型配置  
 MODEL_VERSIONS = {