mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-04-30 19:21:21 +00:00
main_entry.py: add support for new text_encoder_dir cmd arg
copying method used for VAE
This commit is contained in:
@@ -150,6 +150,8 @@ def refresh_models():
|
|||||||
|
|
||||||
if isinstance(shared.cmd_opts.vae_dir, str):
|
if isinstance(shared.cmd_opts.vae_dir, str):
|
||||||
module_paths.append(os.path.abspath(shared.cmd_opts.vae_dir))
|
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:
|
for vae_path in module_paths:
|
||||||
vae_files = find_files_with_extensions(vae_path, file_extensions)
|
vae_files = find_files_with_extensions(vae_path, file_extensions)
|
||||||
|
|||||||
Reference in New Issue
Block a user