From c67a7d0afe113ec0c2067ac9b0b96c13d2164b6d Mon Sep 17 00:00:00 2001 From: liubo0902 <38622806+liubo0902@users.noreply.github.com> Date: Tue, 27 Jun 2023 15:46:53 +0800 Subject: [PATCH 1/2] support lora dir --- scripts/model_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/model_util.py b/scripts/model_util.py index 977eb93..423beea 100644 --- a/scripts/model_util.py +++ b/scripts/model_util.py @@ -310,7 +310,7 @@ def find_closest_lora_model_name(search: str): def update_models(): global lora_models, lora_model_names, legacy_model_names - paths = [lora_models_dir] + paths = [lora_models_dir, shared.cmd_opts.lora_dir] extra_lora_paths = util.split_path_list(shared.opts.data.get("additional_networks_extra_lora_path", "")) for path in extra_lora_paths: path = path.lstrip() From 879859b551fb12357f3f509d33973c61853e608f Mon Sep 17 00:00:00 2001 From: liubo0902 <38622806+liubo0902@users.noreply.github.com> Date: Wed, 28 Jun 2023 19:43:05 +0800 Subject: [PATCH 2/2] Update model_util.py --- scripts/model_util.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scripts/model_util.py b/scripts/model_util.py index 423beea..929905b 100644 --- a/scripts/model_util.py +++ b/scripts/model_util.py @@ -310,7 +310,13 @@ def find_closest_lora_model_name(search: str): def update_models(): global lora_models, lora_model_names, legacy_model_names - paths = [lora_models_dir, shared.cmd_opts.lora_dir] + paths = [lora_models_dir] + if os.path.exists(shared.cmd_opts.lora_dir): + paths.append(shared.cmd_opts.lora_dir) + for folder_path in glob.iglob(os.path.join(shared.cmd_opts.data_dir, '*/models/Lora')): + paths.append(folder_path) + for folder_path in glob.iglob(os.path.join(shared.cmd_opts.data_dir, '*/*/models/Lora')): + paths.append(folder_path) extra_lora_paths = util.split_path_list(shared.opts.data.get("additional_networks_extra_lora_path", "")) for path in extra_lora_paths: path = path.lstrip()