Print key names for easier debugging.

0e06b370 · comfyanonymous · 5eb98f00 · 0e06b370
Commit 0e06b370 authored Jun 14, 2024 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/model_patcher.py comfy/model_patcher.py +1 -1

No files found.
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@@ -335,7 +335,7 @@ class ModelPatcher:
                    self.patch_weight_to_device(bias_key, device_to)
                    m.to(device_to)
                    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.lowvram_patch_counter = patch_counter