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(ui): reset node execution states on cancel

Browse Source
This commit is contained in:
psychedelicious 2023-08-30 18:58:27 +10:00
parent 044d4c107a
commit f2334ec302
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/frontend/web/src/features/nodes/store/nodesSlice.ts
View File

@ -1,5 +1,5 @@
import { createSlice, PayloadAction } from '@reduxjs/toolkit'; import { createSlice, PayloadAction } from '@reduxjs/toolkit';
import { cloneDeep, forEach, isEqual, uniqBy } from 'lodash-es'; import { cloneDeep, forEach, isEqual, map, uniqBy } from 'lodash-es';
import { import {
addEdge, addEdge,
applyEdgeChanges, applyEdgeChanges,
@ -18,7 +18,7 @@ import {
Viewport, Viewport,
} from 'reactflow'; } from 'reactflow';
import { receivedOpenAPISchema } from 'services/api/thunks/schema'; import { receivedOpenAPISchema } from 'services/api/thunks/schema';
import { sessionInvoked } from 'services/api/thunks/session'; import { sessionCanceled, sessionInvoked } from 'services/api/thunks/session';
import { ImageField } from 'services/api/types'; import { ImageField } from 'services/api/types';
import { import {
appSocketGeneratorProgress, appSocketGeneratorProgress,
@ -818,6 +818,13 @@ const nodesSlice = createSlice({
nes.outputs = []; nes.outputs = [];
}); });
}); });
builder.addCase(sessionCanceled.fulfilled, (state) => {
map(state.nodeExecutionStates, (nes) => {
if (nes.status === NodeStatus.IN_PROGRESS) {
nes.status = NodeStatus.PENDING;
}
});
});
}, },
}); });