From 092b5857bbda01be53619e26f42b1fa9b64a2b44 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Wed, 1 Jul 2026 03:09:26 -0500 Subject: [PATCH] refac --- .../components/workspace/Models/ModelEditor.svelte | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/src/lib/components/workspace/Models/ModelEditor.svelte b/src/lib/components/workspace/Models/ModelEditor.svelte index b054678ee8..5c31a03050 100644 --- a/src/lib/components/workspace/Models/ModelEditor.svelte +++ b/src/lib/components/workspace/Models/ModelEditor.svelte @@ -119,8 +119,10 @@ return models .filter( (baseModel) => - (!currentModelId || baseModel.id !== currentModelId) && - !baseModel?.preset && + (!currentModelId || + baseModel.id !== currentModelId || + (edit && baseModel.id === info.base_model_id)) && + (!baseModel?.preset || (edit && baseModel.id === info.base_model_id)) && baseModel?.owned_by !== 'arena' && !(baseModel?.direct ?? false) && ($user?.role === 'admin' || @@ -334,14 +336,16 @@ if (model.base_model_id) { const base_model = $models - .filter((m) => !m?.preset && !(m?.arena ?? false)) + .filter( + (m) => (!m?.preset && !(m?.arena ?? false)) || (edit && m.id === model.base_model_id) + ) .find((m) => [model.base_model_id, `${model.base_model_id}:latest`].includes(m.id)); console.log('base_model', base_model); if (base_model) { model.base_model_id = base_model.id; - } else { + } else if (!(edit && model.base_model_id === model.id)) { model.base_model_id = null; } }