From eaf3e705562d6a4425dafbcf5ed07af9440fdf02 Mon Sep 17 00:00:00 2001 From: admin Date: Sat, 10 Feb 2024 03:40:45 +0800 Subject: [PATCH] [feat] get total_ram from get_total_memory --- 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 282cfb911..3eeee67dc 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -122,7 +122,7 @@ def get_total_memory(dev=None, torch_total_too=False): return mem_total total_vram = get_total_memory(get_torch_device()) / (1024 * 1024) -total_ram = psutil.virtual_memory().total / (1024 * 1024) +total_ram = get_total_memory(torch.device("cpu")) / (1024 * 1024) print("Total VRAM {:0.0f} MB, total RAM {:0.0f} MB".format(total_vram, total_ram)) if not args.normalvram and not args.cpu: if lowvram_available and total_vram <= 4096: