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

Merge f1f4f545f4 into 0904cc3fe5

Browse Source
This commit is contained in:
Buumcode 2026-03-15 21:13:41 +09:00 committed by GitHub
commit bac47130e4
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

@ -726,8 +726,12 @@ class ModelPatcher:
self.pinned.remove(key) self.pinned.remove(key)
def unpin_all_weights(self): def unpin_all_weights(self):
for key in list(self.pinned): if hasattr(self, 'pinned'): # Safety check
self.unpin_weight(key) for key in list(self.pinned):
self.unpin_weight(key)
else:
# Log or skip: self.pinned was not set
pass
def _load_list(self, for_dynamic=False, default_device=None): def _load_list(self, for_dynamic=False, default_device=None):
loading = [] loading = []
@ -1463,8 +1467,14 @@ class ModelPatcher:
return self.model.state_dict_for_saving(unet_state_dict, clip_state_dict=clip_state_dict, vae_state_dict=vae_state_dict, clip_vision_state_dict=clip_vision_state_dict) return self.model.state_dict_for_saving(unet_state_dict, clip_state_dict=clip_state_dict, vae_state_dict=vae_state_dict, clip_vision_state_dict=clip_vision_state_dict)
def __del__(self): def __del__(self):
self.unpin_all_weights() try:
self.detach(unpatch_all=False) 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
class ModelPatcherDynamic(ModelPatcher): class ModelPatcherDynamic(ModelPatcher):