wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-08 10:47:32 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Refactor module_size function.

Browse Source
This commit is contained in:
comfyanonymous 2026-01-05 03:40:49 -05:00 committed by GitHub
parent d1b9822f74
commit 1a30002a41
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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() sd = module.state_dict()
for k in sd: for k in sd:
t = sd[k] t = sd[k]
module_mem += t.nelement() * t.element_size() module_mem += t.nbytes
return module_mem return module_mem
class LoadedModel: class LoadedModel: