Commit 36966e32 authored by Muhammad Rizqi Nur's avatar Muhammad Rizqi Nur
Browse files

Fix #4035

parent d4790fa6
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -201,7 +201,7 @@ def load_model_weights(model, checkpoint_info):

        if shared.opts.sd_checkpoint_cache > 0:
            checkpoints_loaded[checkpoint_info] = model.state_dict().copy()
            while len(checkpoints_loaded) > shared.opts.sd_checkpoint_cache:
            while len(checkpoints_loaded) > shared.opts.sd_checkpoint_cache + 1:
                checkpoints_loaded.popitem(last=False)  # LRU
    else:
        print(f"Loading weights [{sd_model_hash}] from cache")