wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-07 21:00:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Merge f1f4f545f4 into e14f3b6610

Browse Source
This commit is contained in:
Buumcode 2026-01-06 14:37:34 +08:00 committed by GitHub
commit d8ba7238ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
comfy/model_patcher.py
View File

@ -651,8 +651,12 @@ class ModelPatcher:
self.pinned.remove(key)
def unpin_all_weights(self):
for key in list(self.pinned):
self.unpin_weight(key)
if hasattr(self, 'pinned'): # Safety check
for key in list(self.pinned):
self.unpin_weight(key)
else:
# Log or skip: self.pinned was not set
pass
def _load_list(self):
loading = []
@ -1354,6 +1358,12 @@ class ModelPatcher:
self.clear_cached_hook_weights()
def __del__(self):
self.unpin_all_weights()
self.detach(unpatch_all=False)
try:
if hasattr(self, 'pinned'):
self.unpin_all_weights()
self.detach(unpatch_all=False)
except Exception as e:
# Suppress errors in destructor to avoid noise
pass