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

Merge pull request #3826 from ANTONIOPSD/patch-1

Natural sorting for dropdown checkpoint list
parents 810e6a40 5d5dc640
Loading
Loading
Loading
Loading
+5 −2
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ import os.path
import sys
from collections import namedtuple
import torch
import re
from omegaconf import OmegaConf

from ldm.util import instantiate_from_config
@@ -36,7 +37,9 @@ def setup_model():


def checkpoint_tiles(): 
    return sorted([x.title for x in checkpoints_list.values()])
    convert = lambda name: int(name) if name.isdigit() else name.lower() 
    alphanumeric_key = lambda key: [convert(c) for c in re.split('([0-9]+)', key)] 
    return sorted([x.title for x in checkpoints_list.values()], key = alphanumeric_key)


def list_models():