Commit 5d14f282 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fixed a bug where after switching to a checkpoint with unknown hash, you'd get...

fixed a bug where after switching to a checkpoint with unknown hash, you'd get empty space instead of checkpoint name in UI
fixed a bug where if you update a selected checkpoint on disk and then restart the program, a different checkpoint loads, but the name is shown for the the old one.
parent f8feeaae
Loading
Loading
Loading
Loading
+1 −3
Original line number Diff line number Diff line
@@ -231,11 +231,9 @@ def get_checkpoint_state_dict(checkpoint_info: CheckpointInfo, timer):


def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer):
    title = checkpoint_info.title
    sd_model_hash = checkpoint_info.calculate_shorthash()
    timer.record("calculate hash")

    if checkpoint_info.title != title:
    shared.opts.data["sd_model_checkpoint"] = checkpoint_info.title

    if state_dict is None: