diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-18 20:23:09 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-18 20:23:09 +0000 |
| commit | 43fd6eaab8231f68c314b6d4fa41e2ca19582310 (patch) | |
| tree | 3d054acfbfbd01c858912ce88db68b538b227e51 /modules/sd_vae.py | |
| parent | b186d44dcd0df9d127a663b297334a5bd8258b58 (diff) | |
| parent | d906f87043d809e6d4d8de3c9926e184169b330f (diff) | |
| download | stable-diffusion-webui-gfx803-43fd6eaab8231f68c314b6d4fa41e2ca19582310.tar.gz stable-diffusion-webui-gfx803-43fd6eaab8231f68c314b6d4fa41e2ca19582310.tar.bz2 stable-diffusion-webui-gfx803-43fd6eaab8231f68c314b6d4fa41e2ca19582310.zip | |
Merge pull request #6851 from ddPn08/master
Add `--vae-dir` argument
Diffstat (limited to 'modules/sd_vae.py')
| -rw-r--r-- | modules/sd_vae.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/modules/sd_vae.py b/modules/sd_vae.py index b2af2ce7..da1bf15c 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -72,6 +72,13 @@ def refresh_vae_list(): os.path.join(shared.cmd_opts.ckpt_dir, '**/*.vae.safetensors'), ] + if shared.cmd_opts.vae_dir is not None and os.path.isdir(shared.cmd_opts.vae_dir): + paths += [ + os.path.join(shared.cmd_opts.vae_dir, '**/*.ckpt'), + os.path.join(shared.cmd_opts.vae_dir, '**/*.pt'), + os.path.join(shared.cmd_opts.vae_dir, '**/*.safetensors'), + ] + candidates = [] for path in paths: candidates += glob.iglob(path, recursive=True) |