Skip to content

Commit e65e642

Browse files
committed
mm: fix debug message
1 parent 99bed5e commit e65e642

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_management.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -537,6 +537,7 @@ def should_reload_model(self, force_patch_weights=False):
537537
def model_unload(self, memory_to_free=None, unpatch_weights=True):
538538
if self.model is None:
539539
return True
540+
logging.debug(f"Unloading {self.model.model.__class__.__name__}")
540541
if memory_to_free is not None:
541542
if memory_to_free < self.model.loaded_size():
542543
freed, modules_to_offload = self.model.partially_unload(self.model.offload_device, memory_to_free)
@@ -626,7 +627,6 @@ def free_memory(memory_required, device, keep_loaded=[]):
626627
if free_mem > memory_required:
627628
break
628629
memory_to_free = memory_required - free_mem
629-
logging.info(f"Unloading {shift_model.model.model.__class__.__name__}")
630630
if shift_model.model_unload(memory_to_free):
631631
unloaded_model.append((i, shift_model))
632632

0 commit comments

Comments
 (0)