Merge pull request AUTOMATIC1111#13313 from chu8129/dev

use orderdict as lru cache:opt/bug
k1nos · Sep 30, 2023 · cdafbca · cdafbca
2 parents 34055f9 + 8e355fb
commit cdafbca
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modules/sd_models.py b/modules/sd_models.py
@@ -309,6 +309,8 @@ 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")
+        # move to end as latest
+        checkpoints_loaded.move_to_end(checkpoint_info)
         return checkpoints_loaded[checkpoint_info]
 
     print(f"Loading weights [{sd_model_hash}] from {checkpoint_info.filename}")