aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-10-02 17:24:50 +0300
committerAUTOMATIC <16777216c@gmail.com>2022-10-02 17:24:50 +0300
commit0758f6e641b5790ce566a998d43e0ea74a627766 (patch)
treea068a2dd4ef4e8fc182429ea584c2cdbc3661559
parent5f561ee95dcb09d92ea67bab5561dced93fe3d00 (diff)
fix --ckpt option breaking model selection
-rw-r--r--modules/sd_models.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 5b3dbdc7..9259d69e 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -69,7 +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
+ shared.opts.data['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: