Browse Source

Merge pull request #12297 from AUTOMATIC1111/sort-VAE

sort VAE
AUTOMATIC1111 2 years ago
parent
commit
09165916fa
1 changed files with 3 additions and 0 deletions
  1. 3 0
      modules/sd_vae.py

+ 3 - 0
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]