Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Do not add anything but final output to staging area

Browse Source
This commit is contained in:
blessedcoolant 2023-08-12 21:10:30 +12:00
parent f296e5c41e
commit 27bd127fb0
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketInvocationComplete.ts
View File

@ -7,6 +7,7 @@ import {
imageSelected,
} from 'features/gallery/store/gallerySlice';
import { IMAGE_CATEGORIES } from 'features/gallery/store/types';
import { INPAINT_FINAL_IMAGE } from 'features/nodes/util/graphBuilders/constants';
import { progressImageSet } from 'features/system/store/systemSlice';
import { imagesApi } from 'services/api/endpoints/images';
import { isImageOutput } from 'services/api/guards';
@ -52,7 +53,9 @@ export const addInvocationCompleteEventListener = () => {
// Add canvas images to the staging area
if (
graph_execution_state_id === canvas.layerState.stagingArea.sessionId
graph_execution_state_id ===
canvas.layerState.stagingArea.sessionId &&
data.source_node_id === INPAINT_FINAL_IMAGE
) {
dispatch(addImageToStagingArea(imageDTO));
}