From 8c04eec2104a656d9dfa0f98c7cc358237a969bc Mon Sep 17 00:00:00 2001 From: Mary Hipp Date: Fri, 15 Mar 2024 10:12:03 -0400 Subject: [PATCH] fix initial main model logic --- .../listenerMiddleware/listeners/modelsLoaded.ts | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/modelsLoaded.ts b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/modelsLoaded.ts index 52fc2ede4a..2ba9aa3cbf 100644 --- a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/modelsLoaded.ts +++ b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/modelsLoaded.ts @@ -55,14 +55,13 @@ const handleMainModels: ModelHandler = (models, state, dispatch, log) => { 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) { return; } 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) { 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) { log.error({ error: result.error.format() }, 'Failed to parse main model');