This commit is contained in:
Timothy Jaeryang Baek committed 2026-07-01 03:09:26 -05:00
1 parent 0e5540c2b2
commit 092b5857bb
1 file changed
+8 -4
@@ -119,8 +119,10 @@
return models return models
.filter( .filter(
(baseModel) => (baseModel) =>
(!currentModelId || baseModel.id !== currentModelId) && (!currentModelId ||
!baseModel?.preset && baseModel.id !== currentModelId ||
(edit && baseModel.id === info.base_model_id)) &&
(!baseModel?.preset || (edit && baseModel.id === info.base_model_id)) &&
baseModel?.owned_by !== 'arena' && baseModel?.owned_by !== 'arena' &&
!(baseModel?.direct ?? false) && !(baseModel?.direct ?? false) &&
($user?.role === 'admin' || ($user?.role === 'admin' ||
@@ -334,14 +336,16 @@
if (model.base_model_id) { if (model.base_model_id) {
const base_model = $models 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)); .find((m) => [model.base_model_id, `${model.base_model_id}:latest`].includes(m.id));
console.log('base_model', base_model); console.log('base_model', base_model);
if (base_model) { if (base_model) {
model.base_model_id = base_model.id; model.base_model_id = base_model.id;
} else { } else if (!(edit && model.base_model_id === model.id)) {
model.base_model_id = null; model.base_model_id = null;
} }
} }