xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2025-12-16 01:25:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

mm: fix debug message

Browse Source
This commit is contained in:
Rattus 2025-11-27 17:58:39 +10:00
parent 99bed5e19f
commit e65e642fbe
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -537,6 +537,7 @@ class LoadedModel:
def model_unload(self, memory_to_free=None, unpatch_weights=True): def model_unload(self, memory_to_free=None, unpatch_weights=True):
if self.model is None: if self.model is None:
return True return True
logging.debug(f"Unloading {self.model.model.__class__.__name__}")
if memory_to_free is not None: if memory_to_free is not None:
if memory_to_free < self.model.loaded_size(): if memory_to_free < self.model.loaded_size():
freed, modules_to_offload = self.model.partially_unload(self.model.offload_device, memory_to_free) 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=[]):
if free_mem > memory_required: if free_mem > memory_required:
break break
memory_to_free = memory_required - free_mem memory_to_free = memory_required - free_mem
logging.info(f"Unloading {shift_model.model.model.__class__.__name__}")
if shift_model.model_unload(memory_to_free): if shift_model.model_unload(memory_to_free):
unloaded_model.append((i, shift_model)) unloaded_model.append((i, shift_model))