소스 검색

Merge pull request #12297 from AUTOMATIC1111/sort-VAE

sort VAE
AUTOMATIC1111 2 년 전
부모
커밋
09165916fa
1개의 변경된 파일3개의 추가작업 그리고 0개의 파일을 삭제
  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():
 def refresh_vae_list():
+    global vae_dict
     vae_dict.clear()
     vae_dict.clear()
 
 
     paths = [
     paths = [
@@ -83,6 +84,8 @@ def refresh_vae_list():
         name = get_filename(filepath)
         name = get_filename(filepath)
         vae_dict[name] = 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):
 def find_vae_near_checkpoint(checkpoint_file):
     checkpoint_path = os.path.basename(checkpoint_file).rsplit('.', 1)[0]
     checkpoint_path = os.path.basename(checkpoint_file).rsplit('.', 1)[0]