use shallow copy for AUTOMATIC1111#13535

ruchej · Sep 30, 2024 · bc43e16 · bc43e16
1 parent b86c124
commit bc43e16
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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")