mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-03-03 04:11:30 -08:00
Code Issues Projects Releases Packages Wiki Activity

if --ckpt option is specified, load that model

Browse source
This commit is contained in:
AUTOMATIC 2022-09-30 22:16:03 +03:00
parent 3a876b16a9
commit 2b03f0bbda
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/sd_models.py
View file

@ -69,6 +69,7 @@ def list_models():
h = model_hash(cmd_ckpt)
title, short_model_name = modeltitle(cmd_ckpt, h)
checkpoints_list[title] = CheckpointInfo(cmd_ckpt, title, h, short_model_name)
shared.opts.sd_model_checkpoint = title
elif cmd_ckpt is not None and cmd_ckpt != shared.default_sd_model_file:
print(f"Checkpoint in --ckpt argument not found (Possible it was moved to {model_path}: {cmd_ckpt}", file=sys.stderr)
for filename in model_list: