fix: import model

This commit is contained in:
Timothy J. Baek
2024-06-08 19:21:37 -07:00
parent e274a0dc3e
commit 5342063e24
2 changed files with 6 additions and 2 deletions

View File

@@ -127,12 +127,16 @@
if (model.info.base_model_id) {
const base_model = $models
.filter((m) => !m?.preset)
.find((m) => m.id === model.info.base_model_id);
.find((m) =>
[model.info.base_model_id, `${model.info.base_model_id}:latest`].includes(m.id)
);
console.log('base_model', base_model);
if (!base_model) {
model.info.base_model_id = null;
} else if ($models.find((m) => m.id === `${model.info.base_model_id}:latest`)) {
model.info.base_model_id = `${model.info.base_model_id}:latest`;
}
}