Unverified Commit 09165916 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #12297 from AUTOMATIC1111/sort-VAE

sort VAE
parents 3f9e09a6 7f1d087c
Loading
Loading
Loading
Loading
+3 −0
Original line number Original line Diff line number Diff line
@@ -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]