From 28b86d391717bcfb51ac7e4e6f56854ba21cb191 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Thu, 29 Aug 2024 21:37:11 +1000
Subject: [PATCH] fix(ui): incorrect batch origin preventing progress/staging

---
 .../frontend/web/src/services/events/onInvocationComplete.ts    | 2 +-
 invokeai/frontend/web/src/services/events/setEventListeners.tsx | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/invokeai/frontend/web/src/services/events/onInvocationComplete.ts b/invokeai/frontend/web/src/services/events/onInvocationComplete.ts
index 6fcf808aab..03bc860d71 100644
--- a/invokeai/frontend/web/src/services/events/onInvocationComplete.ts
+++ b/invokeai/frontend/web/src/services/events/onInvocationComplete.ts
@@ -154,7 +154,7 @@ export const buildOnInvocationComplete = (
     // Update the node execution states - the image output is handled below
     if (data.origin === 'workflows') {
       await handleOriginWorkflows(data);
-    } else if (data.origin === 'canvas') {
+    } else if (data.origin === 'generation') {
       await handleOriginCanvas(data);
     } else {
       await handleOriginOther(data);
diff --git a/invokeai/frontend/web/src/services/events/setEventListeners.tsx b/invokeai/frontend/web/src/services/events/setEventListeners.tsx
index aff1200bc1..f0aaeb9ad6 100644
--- a/invokeai/frontend/web/src/services/events/setEventListeners.tsx
+++ b/invokeai/frontend/web/src/services/events/setEventListeners.tsx
@@ -131,7 +131,7 @@ export const setEventListeners = ({ socket, dispatch, getState, setIsConnected }
       }
     }
 
-    if (origin === 'canvas' && destination === 'canvas') {
+    if (origin === 'generation' && destination === 'canvas') {
       $lastCanvasProgressEvent.set(data);
     }
   });