Skip to content

Commit b202f84

Browse files
authored
Skip running model finalizers at exit (Comfy-Org#12994)
1 parent 7d5f525 commit b202f84

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

comfy/model_management.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -541,6 +541,7 @@ def _set_model(self, model):
541541
if model.parent is not None:
542542
self._parent_model = weakref.ref(model.parent)
543543
self._patcher_finalizer = weakref.finalize(model, self._switch_parent)
544+
self._patcher_finalizer.atexit = False
544545

545546
def _switch_parent(self):
546547
model = self._parent_model()
@@ -587,6 +588,7 @@ def model_load(self, lowvram_model_memory=0, force_patch_weights=False):
587588

588589
self.real_model = weakref.ref(real_model)
589590
self.model_finalizer = weakref.finalize(real_model, cleanup_models)
591+
self.model_finalizer.atexit = False
590592
return real_model
591593

592594
def should_reload_model(self, force_patch_weights=False):

0 commit comments

Comments
 (0)