From 0379ea2672baafc0e4d5d4ad4411f9e2d1eaa37f Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Wed, 22 May 2024 10:41:27 +0800
Subject: [PATCH] Update Comfyui_CustomNet_node.py

add
---
 Comfyui_CustomNet_node.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Comfyui_CustomNet_node.py b/Comfyui_CustomNet_node.py
index 535222e..48ec99e 100644
--- a/Comfyui_CustomNet_node.py
+++ b/Comfyui_CustomNet_node.py
@@ -2,8 +2,10 @@
 # -*- coding: UTF-8 -*-
 import sys
 import os
+from folder_paths import base_path
+sys.path.append(os.path.join(base_path,"custom_nodes","ComfyUI_CustomNet","custom_net"))
+# print(base_path)
 path = os.path.dirname(os.path.abspath(__file__))
-sys.path.append(os.path.join(path,"custom_net"))
 path_dir = os.path.dirname(path)
 file_path = os.path.dirname(path_dir)
 import numpy as np