fix for broken checkpoint merger

This commit is contained in:
AUTOMATIC 2022-10-19 12:45:30 +03:00
parent 9931c0bd48
commit f894dd552f

View File

@ -148,7 +148,10 @@ def get_state_dict_from_checkpoint(pl_sd):
if new_key is not None:
sd[new_key] = v
return sd
pl_sd.clear()
pl_sd.update(sd)
return pl_sd
def load_model_weights(model, checkpoint_info):