wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 21:30:15 +08:00
Code Issues Actions 12 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
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: