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

Merge f1f4f545f4 into 5943fbf457

Browse Source
This commit is contained in:
Buumcode 2026-01-08 18:41:21 +01:00 committed by GitHub
commit 266d63e705
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 = []
@ -1356,6 +1360,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