Commit eebf4959 authored by cluder's avatar cluder
Browse files

restore #4035 behavior

- if checkpoint cache is set to 1, keep 2 models in cache (current +1 more)
parent f37cce0e
Loading
Loading
Loading
Loading
+1 −1
Original line number Original line Diff line number Diff line
@@ -213,7 +213,7 @@ def load_model_weights(model, checkpoint_info, vae_file="auto"):


    # clean up cache if limit is reached
    # clean up cache if limit is reached
    if cache_enabled:
    if cache_enabled:
        while len(checkpoints_loaded) > shared.opts.sd_checkpoint_cache:
        while len(checkpoints_loaded) > shared.opts.sd_checkpoint_cache + 1: # we need to count the current model
            checkpoints_loaded.popitem(last=False)  # LRU
            checkpoints_loaded.popitem(last=False)  # LRU


    model.sd_model_hash = sd_model_hash
    model.sd_model_hash = sd_model_hash