diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-10-14 05:01:04 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-10-14 05:01:04 +0000 |
| commit | 0619df9835833079f8ba5cb5a510b55c4433acaf (patch) | |
| tree | fd10f339d5d1d83ae50962f635d342b67c5dbe00 /modules | |
| parent | 7cc96429f2f779b2e43804e0c987932cf4c0dd48 (diff) | |
| download | stable-diffusion-webui-gfx803-0619df9835833079f8ba5cb5a510b55c4433acaf.tar.gz stable-diffusion-webui-gfx803-0619df9835833079f8ba5cb5a510b55c4433acaf.tar.bz2 stable-diffusion-webui-gfx803-0619df9835833079f8ba5cb5a510b55c4433acaf.zip | |
use shallow copy for #13535
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 2b43868e..c8efeedc 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -1,5 +1,4 @@ import collections
-import copy
import os.path
import sys
import gc
@@ -360,7 +359,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer if shared.opts.sd_checkpoint_cache > 0:
# cache newly loaded model
- checkpoints_loaded[checkpoint_info] = copy.deepcopy(state_dict)
+ checkpoints_loaded[checkpoint_info] = state_dict.copy()
model.load_state_dict(state_dict, strict=False)
timer.record("apply weights to model")
|