298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 10:53:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Call unpatch_hooks at the start of ModelPatcher.partially_unload

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-03-15 07:37:26 -05:00
parent 55a1b09ddc
commit e03bd9e0e3
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/model_patcher.py
View File

@ -747,6 +747,7 @@ class ModelPatcher:
def partially_unload(self, device_to, memory_to_free=0):
with self.use_ejected():
self.unpatch_hooks()
memory_freed = 0
patch_counter = 0
unload_list = self._load_list()