diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-19 09:45:30 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-19 09:45:42 +0000 |
| commit | f894dd552f68bea27476f1f360ab8e79f3a65b4f (patch) | |
| tree | 5586b58df19727a67e961598c0410ab63fdf15a0 | |
| parent | 9931c0bd48346dc5af23864117becfac33347a7c (diff) | |
| download | stable-diffusion-webui-gfx803-f894dd552f68bea27476f1f360ab8e79f3a65b4f.tar.gz stable-diffusion-webui-gfx803-f894dd552f68bea27476f1f360ab8e79f3a65b4f.tar.bz2 stable-diffusion-webui-gfx803-f894dd552f68bea27476f1f360ab8e79f3a65b4f.zip | |
fix for broken checkpoint merger
| -rw-r--r-- | modules/sd_models.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 7ad6d474..eae22e87 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -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):
|