diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-04 06:38:16 +0000 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-04 06:38:16 +0000 |
| commit | 70e66e81e56f0bb187395878ee49705acbfab40c (patch) | |
| tree | b7745d409c7c9192eec0fe0722acf316f8342f79 /modules/sd_vae.py | |
| parent | c134a480164bef017cd4b33fae57a31a86556beb (diff) | |
| parent | f0c1063a707a4a43823b0ed00e2a8eeb22a9ed0a (diff) | |
| download | stable-diffusion-webui-gfx803-70e66e81e56f0bb187395878ee49705acbfab40c.tar.gz stable-diffusion-webui-gfx803-70e66e81e56f0bb187395878ee49705acbfab40c.tar.bz2 stable-diffusion-webui-gfx803-70e66e81e56f0bb187395878ee49705acbfab40c.zip | |
Merge branch 'dev' into efficient-vae-methods
Diffstat (limited to 'modules/sd_vae.py')
| -rw-r--r-- | modules/sd_vae.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_vae.py b/modules/sd_vae.py index e4ff2994..84271db0 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -50,6 +50,7 @@ def get_filename(filepath): def refresh_vae_list(): + global vae_dict vae_dict.clear() paths = [ @@ -83,6 +84,8 @@ def refresh_vae_list(): name = get_filename(filepath) vae_dict[name] = filepath + vae_dict = dict(sorted(vae_dict.items(), key=lambda item: shared.natural_sort_key(item[0]))) + def find_vae_near_checkpoint(checkpoint_file): checkpoint_path = os.path.basename(checkpoint_file).rsplit('.', 1)[0] |