From 22eb6d6dd092b2ed2e25bdca6938d2011bfde934 Mon Sep 17 00:00:00 2001
From: neverbiasu <1751162157@qq.com>
Date: Tue, 27 Aug 2024 21:00:22 +0800
Subject: [PATCH] Refactor node.py to use project root path for model_dir

---
 node.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/node.py b/node.py
index 46c7eeb..c793fe0 100644
--- a/node.py
+++ b/node.py
@@ -1,9 +1,15 @@
 import os
 import cv2
+import sys
 import torch
 import argparse
 import numpy as np
 
+project_root = os.path.abspath(os.path.dirname(__file__))
+
+if project_root not in sys.path:
+    sys.path.append(project_root)
+
 from PIL import Image
 from diffusers import (
     StableDiffusionControlNetPipeline,
@@ -18,7 +24,7 @@ from annotator.lineart import LineartDetector
 from huggingface_hub import snapshot_download
 from ip_adapter import StyleShot, StyleContentStableDiffusionControlNetPipeline
 
-model_dir = os.path.join(os.path.dirname(__file__), "prtrained_models")
+model_dir = os.path.join(project_root, "prtrained_models")
 device = "cuda"