comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-03 06:57:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor module_size function. (#11637)

Browse Source
This commit is contained in:
comfyanonymous
2026-01-05 00:48:31 -08:00
committed by GitHub
parent d1b9822f74
commit d157c3299d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@@ -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: