diff options
| author | discus0434 <66945496+discus0434@users.noreply.github.com> | 2022-10-19 15:48:13 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-19 15:48:13 +0000 |
| commit | 365d4b1650b77c8f8aedf4a294c2b9e47d665f04 (patch) | |
| tree | 90c719f4b176e8676ea228237f354b73a3d67147 /modules/sd_models.py | |
| parent | 3770b8d2fa62066d472a04739c7b84bce8538832 (diff) | |
| parent | f510a2277ee3641564c13841b04b50293384526f (diff) | |
| download | stable-diffusion-webui-gfx803-365d4b1650b77c8f8aedf4a294c2b9e47d665f04.tar.gz stable-diffusion-webui-gfx803-365d4b1650b77c8f8aedf4a294c2b9e47d665f04.tar.bz2 stable-diffusion-webui-gfx803-365d4b1650b77c8f8aedf4a294c2b9e47d665f04.zip | |
Merge branch 'AUTOMATIC1111:master' into master
Diffstat (limited to 'modules/sd_models.py')
| -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):
|