Refactor module_size function. (#11637)

comfyanonymous · web-flow · commit d157c3299d6f · 2026-01-05T03:48:31.000-05:00
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -456,7 +456,7 @@ def module_size(module):
     sd = module.state_dict()
     for k in sd:
         t = sd[k]
-        module_mem += t.nelement() * t.element_size()
+        module_mem += t.nbytes
     return module_mem
 
 class LoadedModel:
