fix
This commit is contained in:
parent
813535d38b
commit
8e355fbd75
|
@ -309,6 +309,7 @@ def get_checkpoint_state_dict(checkpoint_info: CheckpointInfo, timer):
|
||||||
if checkpoint_info in checkpoints_loaded:
|
if checkpoint_info in checkpoints_loaded:
|
||||||
# use checkpoint cache
|
# use checkpoint cache
|
||||||
print(f"Loading weights [{sd_model_hash}] from cache")
|
print(f"Loading weights [{sd_model_hash}] from cache")
|
||||||
|
# move to end as latest
|
||||||
checkpoints_loaded.move_to_end(checkpoint_info)
|
checkpoints_loaded.move_to_end(checkpoint_info)
|
||||||
return checkpoints_loaded[checkpoint_info]
|
return checkpoints_loaded[checkpoint_info]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue