fix: show all model version

This commit is contained in:
BennyKok 2024-01-14 00:53:42 +08:00
parent 6c49bff095
commit b828e37ed3

View File

@ -185,18 +185,19 @@ function mapModelsList(
models: z.infer<typeof CivitalModelSchema> models: z.infer<typeof CivitalModelSchema>
): z.infer<typeof ModelListWrapper> { ): z.infer<typeof ModelListWrapper> {
return { return {
models: models.items.map((item) => { models: models.items.flatMap((item) => {
const v = item.modelVersions[0]; return item.modelVersions.map((v) => {
return { return {
name: `${item.name} ${v.name} (${v.files[0].name})`, name: `${item.name} ${v.name} (${v.files[0].name})`,
type: mapType(item.type.toLowerCase()), type: mapType(item.type.toLowerCase()),
base: v.baseModel, base: v.baseModel,
save_path: "default", save_path: "default",
description: item.description, description: item.description,
reference: "", reference: "",
filename: v.files[0].name, filename: v.files[0].name,
url: v.files[0].downloadUrl, url: v.files[0].downloadUrl,
} as z.infer<typeof Model>; } as z.infer<typeof Model>;
});
}), }),
}; };
} }