main_entry.py: add support for new text_encoder_dir cmd arg

copying method used for VAE
This commit is contained in:
DenOfEquity
2024-08-25 15:01:30 +01:00
committed by GitHub
parent 59dd981fa7
commit 8c6b64e6e9

View File

@@ -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)