From 8c6b64e6e98ea726b8272e63ecaf12d22dd811f0 Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Sun, 25 Aug 2024 15:01:30 +0100 Subject: [PATCH] main_entry.py: add support for new text_encoder_dir cmd arg copying method used for VAE --- modules_forge/main_entry.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules_forge/main_entry.py b/modules_forge/main_entry.py index 71164bb3..7ac7b853 100644 --- a/modules_forge/main_entry.py +++ b/modules_forge/main_entry.py @@ -150,6 +150,8 @@ def refresh_models(): if isinstance(shared.cmd_opts.vae_dir, str): module_paths.append(os.path.abspath(shared.cmd_opts.vae_dir)) + if isinstance(shared.cmd_opts.text_encoder_dir, str): + module_paths.append(os.path.abspath(shared.cmd_opts.text_encoder_dir)) for vae_path in module_paths: vae_files = find_files_with_extensions(vae_path, file_extensions)