Skip to content

Commit a86684c

Browse files
Merge branch 'master' into ops-changes
2 parents 101644c + d157c32 commit a86684c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_management.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,7 @@ def module_size(module):
456456
sd = module.state_dict()
457457
for k in sd:
458458
t = sd[k]
459-
module_mem += t.nelement() * t.element_size()
459+
module_mem += t.nbytes
460460
return module_mem
461461

462462
class LoadedModel:

0 commit comments

Comments
 (0)