diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-04 08:53:19 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-04 08:53:19 +0000 |
| commit | c93857922ae2a97ede8a98f5281b36201146f074 (patch) | |
| tree | 3716bdb0aec6fee06e2a47d952c6bd0fb874eb73 | |
| parent | 6391128b415b2ba9790f02cdd4863d160584111b (diff) | |
| parent | 24f21583cdba2ae6cc51773b956c6ce068d3dfe4 (diff) | |
| download | stable-diffusion-webui-gfx803-c93857922ae2a97ede8a98f5281b36201146f074.tar.gz stable-diffusion-webui-gfx803-c93857922ae2a97ede8a98f5281b36201146f074.tar.bz2 stable-diffusion-webui-gfx803-c93857922ae2a97ede8a98f5281b36201146f074.zip | |
Merge pull request #12201 from AnyISalIn/dev
fix: sdxl model invalid configuration after the hijack
| -rw-r--r-- | modules/sd_models.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 1d93d893..ba15b451 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -303,12 +303,13 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer sd_models_xl.extend_sdxl(model)
model.load_state_dict(state_dict, strict=False)
- del state_dict
timer.record("apply weights to model")
if shared.opts.sd_checkpoint_cache > 0:
# cache newly loaded model
- checkpoints_loaded[checkpoint_info] = model.state_dict().copy()
+ checkpoints_loaded[checkpoint_info] = state_dict
+
+ del state_dict
if shared.cmd_opts.opt_channelslast:
model.to(memory_format=torch.channels_last)
|