More accurate total

2023-06-02 00:14:41 -05:00 · 2023-06-02 00:14:41 -05:00 · 499641ebf1
parent b5dd15c67a
commit 499641ebf1
1 changed files with 1 additions and 1 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -328,7 +328,7 @@ def get_total_memory(dev=None, torch_total_too=False):
            mem_reserved = stats['reserved_bytes.all.current']
            _, mem_total_cuda = torch.cuda.mem_get_info(dev)
            mem_total_torch = mem_reserved
-            mem_total = mem_total_cuda + mem_total_torch
+            mem_total = mem_total_cuda

    if torch_total_too:
        return (mem_total, mem_total_torch)