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: