Print key names for easier debugging.

This commit is contained in:
comfyanonymous 2024-06-14 18:18:53 -04:00
parent 5eb98f0092
commit 0e06b370db

View File

@ -335,7 +335,7 @@ class ModelPatcher:
self.patch_weight_to_device(bias_key, device_to) self.patch_weight_to_device(bias_key, device_to)
m.to(device_to) m.to(device_to)
mem_counter += comfy.model_management.module_size(m) mem_counter += comfy.model_management.module_size(m)
logging.debug("lowvram: loaded module regularly {}".format(m)) logging.debug("lowvram: loaded module regularly {} {}".format(n, m))
self.model_lowvram = True self.model_lowvram = True
self.lowvram_patch_counter = patch_counter self.lowvram_patch_counter = patch_counter