From e237b6e64f50fc07143d3271be316933e1abecaa Mon Sep 17 00:00:00 2001 From: nullquant Date: Mon, 11 Nov 2024 19:15:48 +0300 Subject: [PATCH] bug in folder reading --- brushnet_nodes.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/brushnet_nodes.py b/brushnet_nodes.py index 2eaa366..47803d0 100644 --- a/brushnet_nodes.py +++ b/brushnet_nodes.py @@ -569,12 +569,8 @@ def get_files_with_extension(folder_name, extension=['.safetensors']): folders = [] if not folders: - folders = [os.path.join(folder_paths.models_dir, folder_name)] - if not os.path.isdir(folders[0]): - folders = [os.path.join(folder_paths.base_path, folder_name)] - if not os.path.isdir(folders[0]): - return {} - + folders = [os.path.join(folder_paths.models_dir, folder_name), os.path.join(folder_paths.base_path, folder_name)] + filtered_folders = [] for x in folders: if not os.path.isdir(x):