From 201ca283005806fdf7e6ec836a11dd4e4c430f26 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 30 Sep 2025 14:27:43 -0400 Subject: [PATCH] improve ram monitor Signed-off-by: Vladimir Mandic --- modules/memstats.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/modules/memstats.py b/modules/memstats.py index fdc94df37..1bb83998b 100644 --- a/modules/memstats.py +++ b/modules/memstats.py @@ -54,11 +54,24 @@ def ram_stats(): ram_total = 100 * res.rss / process.memory_percent() ram_total = min(ram_total, get_docker_limit(), get_runpod_limit()) ram['total'] = gb(ram_total) - ram['used'] = gb(res.rss) - ram['free'] = round(ram['total'] - ram['used']) + ram['rss'] = gb(res.rss) except Exception as e: ram['total'] = 0 + ram['rss'] = 0 + ram['error'] = str(e) + if not fail_once: + shared.log.error(f'RAM stats: {e}') + errors.display(e, 'RAM stats') + fail_once = True + try: + vmem = psutil.virtual_memory() + ram['used'] = gb(vmem.used) + ram['free'] = gb(vmem.free) + ram['avail'] = gb(vmem.available) + except Exception as e: ram['used'] = 0 + ram['free'] = 0 + ram['avail'] = 0 ram['error'] = str(e) if not fail_once: shared.log.error(f'RAM stats: {e}')