From 499641ebf1be190e20624ee352e9dc88884e3df1 Mon Sep 17 00:00:00 2001
From: space-nuko <24979496+space-nuko@users.noreply.github.com>
Date: Fri, 2 Jun 2023 00:14:41 -0500
Subject: [PATCH] More accurate total

---
 comfy/model_management.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 3b7b1dbf..0ea0c71e 100644
--- 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)