From 46db1405df64db4c543b7cc9db958479a3db200f Mon Sep 17 00:00:00 2001 From: EyeDeck Date: Sun, 18 Sep 2022 07:03:17 -0400 Subject: [PATCH] ...and make sure it can't get stuck on in case someone sets opts.memmon_poll_rate to 0 during generation --- modules/ui.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules/ui.py b/modules/ui.py index 8ca3e87f..451ad253 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -119,7 +119,8 @@ def save_files(js_data, images, index): def wrap_gradio_call(func): def f(*args, **kwargs): - if opts.memmon_poll_rate > 0 and not shared.mem_mon.disabled: + run_memmon = opts.memmon_poll_rate > 0 and not shared.mem_mon.disabled + if run_memmon: shared.mem_mon.monitor() t = time.perf_counter() @@ -137,7 +138,7 @@ def wrap_gradio_call(func): elapsed = time.perf_counter() - t - if opts.memmon_poll_rate > 0 and not shared.mem_mon.disabled: + if run_memmon: mem_stats = {k: -(v//-(1024*1024)) for k, v in shared.mem_mon.stop().items()} active_peak = mem_stats['active_peak'] reserved_peak = mem_stats['reserved_peak']