298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

More accurate total

Browse Source
This commit is contained in:
space-nuko 2023-06-02 00:14:41 -05:00
parent b5dd15c67a
commit 499641ebf1
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

@ -328,7 +328,7 @@ def get_total_memory(dev=None, torch_total_too=False):
mem_reserved = stats['reserved_bytes.all.current'] mem_reserved = stats['reserved_bytes.all.current']
_, mem_total_cuda = torch.cuda.mem_get_info(dev) _, mem_total_cuda = torch.cuda.mem_get_info(dev)
mem_total_torch = mem_reserved mem_total_torch = mem_reserved
mem_total = mem_total_cuda + mem_total_torch mem_total = mem_total_cuda
if torch_total_too: if torch_total_too:
return (mem_total, mem_total_torch) return (mem_total, mem_total_torch)