update memmon

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/call_queue.py

@@ -74,28 +74,8 @@ def wrap_gradio_call(func, extra_outputs=None, add_stats=False, name=None):
         elapsed_s = elapsed % 60
         elapsed_text = f"{elapsed_m}m {elapsed_s:.2f}s" if elapsed_m > 0 else f"{elapsed_s:.2f}s"
         summary = timer.process.summary(min_time=0.25, total=False).replace('=', ' ')
-        gpu = ''
-        cpu = ''
-        if not shared.mem_mon.disabled:
-            mem_mon_read = shared.mem_mon.read()
-            ooms = mem_mon_read.pop("oom")
-            retries = mem_mon_read.pop("retries")
-            vram = {k: v//1048576 for k, v in mem_mon_read.items()}
-            if 'active_peak' in vram:
-                peak = max(vram['active_peak'], vram['reserved_peak'], vram['used'])
-                used = round(100.0 * peak / vram['total']) if vram['total'] > 0 else 0
-            else:
-                peak = 0
-                used = 0
-            if peak > 0:
-                gpu += f"| GPU {peak} MB"
-                gpu += f" {used}%" if used > 0 else ''
-                gpu += f" | retries {retries} oom {ooms}" if retries > 0 or ooms > 0 else ''
-        ram = shared.ram_stats()
-        if ram['used'] > 0:
-            cpu += f"| RAM {ram['used']} GB"
-            cpu += f" {round(100.0 * ram['used'] / ram['total'])}%" if ram['total'] > 0 else ''
+        memory = shared.mem_mon.summary()
         if isinstance(res, list) and isinstance(res[-1], str):
-            res[-1] += f"<div class='performance'><p>Time: {elapsed_text} | {summary} {gpu} {cpu}</p></div>"
+            res[-1] += f"<div class='performance'><p>Time: {elapsed_text} | {summary} {memory}</p></div>"
         return tuple(res)
     return f