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 initial main model logic

Browse Source
This commit is contained in:
Mary Hipp 2024-03-15 10:12:03 -04:00 committed by Mary Hipp Rogers
parent 330e1354b4
commit 8c04eec210
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/modelsLoaded.ts
View File

@ -55,14 +55,13 @@ const handleMainModels: ModelHandler = (models, state, dispatch, log) => {
return; return;
} }
const isCurrentMainModelAvailable = currentModel ? models.some((m) => m.key === currentModel.key) : false; const isCurrentMainModelAvailable = currentModel ? mainModels.some((m) => m.key === currentModel.key) : false;
if (isCurrentMainModelAvailable) { if (isCurrentMainModelAvailable) {
return; return;
} }
const defaultModel = state.config.sd.defaultModel; const defaultModel = state.config.sd.defaultModel;
const defaultModelInList = defaultModel ? models.find((m) => m.key === defaultModel) : false; const defaultModelInList = defaultModel ? mainModels.find((m) => m.key === defaultModel) : false;
if (defaultModelInList) { if (defaultModelInList) {
const result = zParameterModel.safeParse(defaultModelInList); const result = zParameterModel.safeParse(defaultModelInList);
@ -84,7 +83,7 @@ const handleMainModels: ModelHandler = (models, state, dispatch, log) => {
} }
} }
const result = zParameterModel.safeParse(models[0]); const result = zParameterModel.safeParse(mainModels[0]);
if (!result.success) { if (!result.success) {
log.error({ error: result.error.format() }, 'Failed to parse main model'); log.error({ error: result.error.format() }, 'Failed to parse main model');