More accurate total

499641eb · space-nuko · b5dd15c6 · 499641eb
Commit 499641eb authored Jun 02, 2023 by space-nuko
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/model_management.py comfy/model_management.py +1 -1

No files found.
--- 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)