Skip to content

Commit e632e5d

Browse files
Add logging for model unloading. (#10692)
1 parent 2abd2b5 commit e632e5d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

comfy/model_patcher.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -909,6 +909,7 @@ def partially_unload(self, device_to, memory_to_free=0):
909909
self.model.model_lowvram = True
910910
self.model.lowvram_patch_counter += patch_counter
911911
self.model.model_loaded_weight_memory -= memory_freed
912+
logging.info("loaded partially: {:.2f} MB loaded, lowvram patches: {}".format(self.model.model_loaded_weight_memory / (1024 * 1024), self.model.lowvram_patch_counter))
912913
return memory_freed
913914

914915
def partially_load(self, device_to, extra_memory=0, force_patch_weights=False):

0 commit comments

Comments
 (0)