Merge pull request #13313 from chu8129/dev

use orderdict as lru cache:opt/bug
2023-09-30 08:55:54 +03:00 · 2023-09-30 08:55:54 +03:00 · cdafbcaad2
parent 34055f9d0c 8e355fbd75
commit cdafbcaad2
1 changed files with 2 additions and 0 deletions
--- 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}")