From df0fee9396ca523b4309f0d49bdf798a21524e5f Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Wed, 14 Aug 2024 17:58:12 -0700 Subject: [PATCH] maybe solve --vae-path --- modules_forge/main_entry.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/modules_forge/main_entry.py b/modules_forge/main_entry.py index c5d6953e..3645f247 100644 --- a/modules_forge/main_entry.py +++ b/modules_forge/main_entry.py @@ -133,17 +133,21 @@ def refresh_models(): shared_items.refresh_checkpoints() ckpt_list = shared_items.list_checkpoint_tiles(shared.opts.sd_checkpoint_dropdown_use_short) - vae_path = os.path.abspath(os.path.join(paths.models_path, "VAE")) - text_encoder_path = os.path.abspath(os.path.join(paths.models_path, "text_encoder")) file_extensions = ['ckpt', 'pt', 'bin', 'safetensors'] module_list.clear() + + module_paths = [ + os.path.abspath(os.path.join(paths.models_path, "VAE")), + os.path.abspath(os.path.join(paths.models_path, "text_encoder")), + ] - vae_files = find_files_with_extensions(vae_path, file_extensions) - module_list.update(vae_files) + if isinstance(shared.cmd_opts.vae_dir, str): + module_paths.append(os.path.abspath(shared.cmd_opts.vae_dir)) - text_encoder_files = find_files_with_extensions(text_encoder_path, file_extensions) - module_list.update(text_encoder_files) + for vae_path in module_paths: + vae_files = find_files_with_extensions(vae_path, file_extensions) + module_list.update(vae_files) return ckpt_list, module_list.keys()