Add job argument to State.begin()

2025-08-04 11:12:35 +00:00 · 2023-06-30 13:11:31 +03:00
parent b70001e618
commit f44feb6a10
6 changed files with 13 additions and 16 deletions
--- a/modules/call_queue.py
+++ b/modules/call_queue.py
@@ -30,7 +30,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
            id_task = None

        with queue_lock:
-            shared.state.begin()
+            shared.state.begin(job=id_task)
            progress.start_task(id_task)

            try: