diff options
| -rw-r--r-- | modules/ui.py | 5 |
1 files 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']
|