store the last generated result

2026-09-19 17:25:09 +02:00 · 2023-02-06 03:53:05 +08:00
parent dbca512154
commit 9407f1731a
2 changed files with 11 additions and 0 deletions
@@ -37,6 +37,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
                res = func(*args, **kwargs)
            finally:
                progress.finish_task(id_task)
+                progress.set_last_task_result(id_task, res)

            shared.state.end()