mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-08 15:17:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Try to improve memory issues with del.

Browse Source
This commit is contained in:
comfyanonymous 2023-02-28 12:27:43 -05:00
parent cd85f876f2
commit b31daadc03
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/sd.py
View File

@ -222,8 +222,11 @@ class ModelPatcher:
return self.model return self.model
def unpatch_model(self): def unpatch_model(self):
model_sd = self.model.state_dict() model_sd = self.model.state_dict()
for k in self.backup: keys = list(self.backup.keys())
for k in keys:
model_sd[k][:] = self.backup[k] model_sd[k][:] = self.backup[k]
del self.backup[k]
self.backup = {} self.backup = {}
def load_lora_for_models(model, clip, lora_path, strength_model, strength_clip): def load_lora_for_models(model, clip, lora_path, strength_model, strength_clip):