...and make sure it can't get stuck on

in case someone sets opts.memmon_poll_rate to 0 during generation

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']