StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix for broken checkpoint merger

Browse Source
This commit is contained in:
AUTOMATIC 2022-10-19 12:45:30 +03:00
parent 9931c0bd48
commit f894dd552f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/sd_models.py
View File

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