fix: prevent getting jobs when generation running (#90)

* fix: prevent getting jobs when generation running * fix: only lock job getting
sdwebui-w-horde · Mar 24, 2023 · a3e67b5 · a3e67b5
1 parent 2372bc1
commit a3e67b5
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/stable_horde/horde.py b/stable_horde/horde.py
@@ -188,11 +188,14 @@ async def run(self):
 
             if self.config.enabled:
                 try:
-                    req = await HordeJob.get(
-                        await self.get_session(),
-                        self.config,
-                        list(self.current_models.keys()),
-                    )
+                    # Require a queue lock to prevent getting jobs when
+                    # there are generation jobs from webui.
+                    with call_queue.queue_lock:
+                        req = await HordeJob.get(
+                            await self.get_session(),
+                            self.config,
+                            list(self.current_models.keys()),
+                        )
                     if req is None:
                         continue