github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(nodes): fix cancel_by_batch_ids result count

Browse Source
This commit is contained in:
psychedelicious
2023-09-14 22:53:49 +10:00
parent 916404745c
commit 77150ab7cd
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/services/session_queue/session_queue_sqlite.py
View File

@@ -409,10 +409,11 @@ class SqliteSessionQueue(SessionQueueBase):
try:
self._lock.acquire()
placeholders = ", ".join(["?" for _ in batch_ids])
where = f"WHERE batch_id IN ({placeholders}) AND status != 'canceled' AND status != 'completed'"
self._cursor.execute(
f"""--sql
SELECT COUNT(*) FROM session_queue
WHERE batch_id IN ({placeholders});
{where};
""",
batch_ids,
)
@@ -421,7 +422,7 @@ class SqliteSessionQueue(SessionQueueBase):
f"""--sql
UPDATE session_queue
SET status = 'canceled'
WHERE batch_id IN ({placeholders}) AND status != 'canceled' AND status != 'completed';
{where};
""",
batch_ids,
)