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(ui): simplify model install event listeners

Browse Source
This commit is contained in:
psychedelicious
2024-02-24 00:17:37 +11:00
parent 7c41b3439a
commit 62c67d7c4b
1 changed files with 16 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketModelInstall.ts
View File

@@ -15,13 +15,12 @@ export const addModelInstallEventListener = () => {
dispatch(
modelsApi.util.updateQueryData('getModelImports', undefined, (draft) => {
const models = JSON.parse(JSON.stringify(draft))
const modelIndex = models.findIndex((m) => m.id === id);
models[modelIndex].bytes = bytes;
models[modelIndex].status = 'downloading';
return models;
const modelImport = draft.find((m) => m.id === id);
if (modelImport) {
modelImport.bytes = bytes;
modelImport.status = 'downloading';
}
return draft;
})
);
},
@@ -34,12 +33,11 @@ export const addModelInstallEventListener = () => {
dispatch(
modelsApi.util.updateQueryData('getModelImports', undefined, (draft) => {
const models = JSON.parse(JSON.stringify(draft))
const modelIndex = models.findIndex((m) => m.id === id);
models[modelIndex].status = 'completed';
return models;
const modelImport = draft.find((m) => m.id === id);
if (modelImport) {
modelImport.status = 'completed';
}
return draft;
})
);
},
@@ -52,12 +50,11 @@ export const addModelInstallEventListener = () => {
dispatch(
modelsApi.util.updateQueryData('getModelImports', undefined, (draft) => {
const models = JSON.parse(JSON.stringify(draft))
const modelIndex = models.findIndex((m) => m.id === id);
models[modelIndex].status = 'error';
return models;
const modelImport = draft.find((m) => m.id === id);
if (modelImport) {
modelImport.status = 'error';
}
return draft;
})
);
},