aboutsummaryrefslogtreecommitdiff
path: root/modules
diff options
context:
space:
mode:
authorqiuwen.wang <chu8129@users.noreply.github.com>2023-09-15 18:23:23 +0800
committerGitHub <noreply@github.com>2023-09-15 18:23:23 +0800
commit813535d38bbcdd8ccc51d0618a7d9fd353677bb9 (patch)
tree383bc88485ef3481130d96dc2432198045f37e43 /modules
parent5ef669de080814067961f28357256e8fe27544f4 (diff)
use dict[key]=model; did not update orderdict order, should use move to end
Diffstat (limited to 'modules')
-rw-r--r--modules/sd_models.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 930d0bee..6d17dd3c 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -309,6 +309,7 @@ def get_checkpoint_state_dict(checkpoint_info: CheckpointInfo, timer):
if checkpoint_info in checkpoints_loaded:
# use checkpoint cache
print(f"Loading weights [{sd_model_hash}] from cache")
+ checkpoints_loaded.move_to_end(checkpoint_info)
return checkpoints_loaded[checkpoint_info]
print(f"Loading weights [{sd_model_hash}] from {checkpoint_info.filename}")