hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update lists of models after merging them in checkpoints tab

Browse Source 
support saving as half
This commit is contained in:
AUTOMATIC
2022-09-29 00:59:44 +03:00
parent 0dc904aa3d
commit 7acfaca05a
4 changed files with 52 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
modules/sd_models.py
View File

@@ -23,6 +23,11 @@ except Exception:
pass
def checkpoint_tiles():
print(sorted([x.title for x in checkpoints_list.values()]))
return sorted([x.title for x in checkpoints_list.values()])
def list_models():
checkpoints_list.clear()
@@ -39,13 +44,14 @@ def list_models():
if name.startswith("\\") or name.startswith("/"):
name = name[1:]
return f'{name} [{h}]'
shortname = os.path.splitext(name.replace("/", "_").replace("\\", "_"))[0]
return f'{name} [{h}]', shortname
cmd_ckpt = shared.cmd_opts.ckpt
if os.path.exists(cmd_ckpt):
h = model_hash(cmd_ckpt)
title = modeltitle(cmd_ckpt, h)
model_name = title.rsplit(".",1)[0] # remove extension if present
title, model_name = modeltitle(cmd_ckpt, h)
checkpoints_list[title] = CheckpointInfo(cmd_ckpt, title, h, model_name)
elif cmd_ckpt is not None and cmd_ckpt != shared.default_sd_model_file:
print(f"Checkpoint in --ckpt argument not found: {cmd_ckpt}", file=sys.stderr)
@@ -53,8 +59,7 @@ def list_models():
if os.path.exists(model_dir):
for filename in glob.glob(model_dir + '/**/*.ckpt', recursive=True):
h = model_hash(filename)
title = modeltitle(filename, h)
model_name = title.rsplit(".",1)[0] # remove extension if present
title, model_name = modeltitle(filename, h)
checkpoints_list[title] = CheckpointInfo(filename, title, h, model_name)