From cfbbf1da31dbcdd44a593ceb514e36840bac06a7 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sat, 1 Mar 2025 17:56:18 +0800
Subject: [PATCH] Update PhotoDoodle_node.py

---
 PhotoDoodle_node.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/PhotoDoodle_node.py b/PhotoDoodle_node.py
index 5be8285..aaef62b 100644
--- a/PhotoDoodle_node.py
+++ b/PhotoDoodle_node.py
@@ -54,18 +54,17 @@ class PhotoDoodle_Loader:
             pre_lora_path = folder_paths.get_full_path("loras", pre_lora)
         else:
             raise ValueError("No model selected")
-        
-        if flux_unet != "none":
-            flux_transformer_path = folder_paths.get_full_path("diffusion_models", flux_unet)
-        else:
-            raise ValueError("No model selected")
-        
+    
         if loras != "none":
             lora_path = folder_paths.get_full_path("loras", loras)
         else:
             raise ValueError("No model selected")
         need_clip=False
         if  not flux_repo:
+            if flux_unet != "none":
+                flux_transformer_path = folder_paths.get_full_path("diffusion_models", flux_unet)
+            else:
+                raise ValueError("No model selected")
             if vae != "none":
                 vae_path = folder_paths.get_full_path("vae", vae)
                 vae_config=os.path.join(flux_repo_local, 'vae')