store the last generated result

2025-12-19 17:47:25 -05:00 · 2023-02-06 03:53:05 +08:00
parent dbca512154
commit 9407f1731a
2 changed files with 11 additions and 0 deletions
--- a/modules/call_queue.py
+++ b/modules/call_queue.py
@@ -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()