Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-17 08:19:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4502 from awsr/ext-compat1

Browse Source 
Add short version of checkpoint titles
This commit is contained in:
Vladimir Mandic
2025-12-27 08:50:02 +01:00
committed by GitHub
parent 448e7b7735 2e1e141a39
commit 48dbda3e30
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_checkpoint.py
+6 -2
View File
@@ -116,11 +116,15 @@ def setup_model():
# sd_hijack_accelerate.hijack_torch_conv()
def checkpoint_titles():
def checkpoint_titles(use_short=False):
def convert(name):
return int(name) if name.isdigit() else name.lower()
def alphanumeric_key(key):
return [convert(c) for c in re.split('([0-9]+)', key)]
return [convert(c) for c in re.split("([0-9]+)", key)]
if use_short:
return sorted([x.title.rsplit("\\", 1)[-1].rsplit("/", 1)[-1] for x in checkpoints_list.values()], key=alphanumeric_key)
return sorted([x.title for x in checkpoints_list.values()], key=alphanumeric_key)