From d2d18d08d35c5efdfbff7469849c7b74d1a71ca2 Mon Sep 17 00:00:00 2001
From: Gius <112352961+GiusTex@users.noreply.github.com>
Date: Thu, 15 May 2025 17:23:00 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index c92cb9c..a4ee500 100644
--- a/nodes.py
+++ b/nodes.py
@@ -213,8 +213,8 @@ class LoadDiffuserModel:
 
         # Go 2 folders back
         comfy_dir = os.path.dirname(os.path.dirname(my_dir))
-        unet_path = f"D:/models/diffusion_models/{unet_name}"
-
+        unet_path = folder_paths.get_full_path_or_raise("diffusion_models", unet_name)
+        
         device = get_device_by_name(device)
         dtype = get_dtype_by_name(dtype)
         
@@ -256,8 +256,8 @@ class LoadDiffuserControlnet:
 
         # Go 2 folders back
         comfy_dir = os.path.dirname(os.path.dirname(my_dir))
-        #controlnet_path = f"D:/models/controlnet/{controlnet_model}"  <-----------FIX
-
+        controlnet_path = folder_paths.get_full_path_or_raise("controlnet", controlnet_model)
+        
         device = get_device_by_name(device)
         dtype = get_dtype_by_name(dtype)