From 7eebd59c692cc423bcede0f56ee2b81c1248c5c6 Mon Sep 17 00:00:00 2001 From: bigcat88 Date: Sun, 15 Jun 2025 08:30:00 +0300 Subject: [PATCH] use cgroup in "total_ram" report Signed-off-by: bigcat88 --- 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 edea5aede..3100b72a0 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -241,7 +241,7 @@ def _get_cgroup_value(p: Path): _CGROUP_LIMIT_BYTES = _get_cgroup_value(_LIMIT_F) total_vram = get_total_memory(get_torch_device()) / (1024 * 1024) -total_ram = psutil.virtual_memory().total / (1024 * 1024) +total_ram = (_CGROUP_LIMIT_BYTES or psutil.virtual_memory().total) / (1024 * 1024) logging.info("Total VRAM {:0.0f} MB, total RAM {:0.0f} MB".format(total_vram, total_ram)) try: