alternate restore progress button implementation

This commit is contained in:
AUTOMATIC
2023-04-29 22:16:54 +03:00
parent bd9700405a
commit c48ab36cb9
6 changed files with 99 additions and 7 deletions

View File

@@ -35,6 +35,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
try:
res = func(*args, **kwargs)
progress.record_results(id_task, res)
finally:
progress.finish_task(id_task)