Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
add base_model field
Browse files
src/routes/api/models/+server.ts
CHANGED
@@ -93,8 +93,8 @@ export async function PATCH({ request } : RequestEvent) {
|
|
93 |
continue;
|
94 |
}
|
95 |
|
96 |
-
const base_model = hugging_face_model?.tags?.find((tag: string) => tag.startsWith("base_model:"))?.split(":")[1] ?? null
|
97 |
-
|
98 |
await prisma.model.update({
|
99 |
where: {
|
100 |
id: model.id
|
|
|
93 |
continue;
|
94 |
}
|
95 |
|
96 |
+
const base_model = hugging_face_model?.[0]?.tags?.find((tag: string) => tag.startsWith("base_model:"))?.split(":")[1] ?? null
|
97 |
+
|
98 |
await prisma.model.update({
|
99 |
where: {
|
100 |
id: model.id
|