diff --git a/src/lib/components/admin/Settings/Models.svelte b/src/lib/components/admin/Settings/Models.svelte index 990b5969d6..a6c96a1298 100644 --- a/src/lib/components/admin/Settings/Models.svelte +++ b/src/lib/components/admin/Settings/Models.svelte @@ -506,6 +506,10 @@ return null; }); + if (res) { + savedModels = [...savedModels, res]; + } + if (res && showToast) { toast.success($i18n.t('Model updated successfully')); await init().catch((error) => toast.error(`${error}`)); @@ -515,8 +519,8 @@ }; const toggleModelHandler = async (model) => { - if (!Object.keys(model).includes('base_model_id') && !isPresetModel(model)) { - await createNewModel(localStorage.token, { + if (!savedModels.find((m: ModelListItem) => m.id === model.id) && !isPresetModel(model)) { + const res = await createNewModel(localStorage.token, { id: model.id, name: model.name, base_model_id: null, @@ -527,6 +531,10 @@ }).catch((error) => { return null; }); + + if (res) { + savedModels = [...savedModels, res]; + } } else { await toggleModelById(localStorage.token, model.id); }