diff --git a/web/src/components/custom-form/ModelPickerView.tsx b/web/src/components/custom-form/ModelPickerView.tsx index 92c8055..b8bb240 100644 --- a/web/src/components/custom-form/ModelPickerView.tsx +++ b/web/src/components/custom-form/ModelPickerView.tsx @@ -185,18 +185,19 @@ function mapModelsList( models: z.infer ): z.infer { return { - models: models.items.map((item) => { - const v = item.modelVersions[0]; - return { - name: `${item.name} ${v.name} (${v.files[0].name})`, - type: mapType(item.type.toLowerCase()), - base: v.baseModel, - save_path: "default", - description: item.description, - reference: "", - filename: v.files[0].name, - url: v.files[0].downloadUrl, - } as z.infer; + models: models.items.flatMap((item) => { + return item.modelVersions.map((v) => { + return { + name: `${item.name} ${v.name} (${v.files[0].name})`, + type: mapType(item.type.toLowerCase()), + base: v.baseModel, + save_path: "default", + description: item.description, + reference: "", + filename: v.files[0].name, + url: v.files[0].downloadUrl, + } as z.infer; + }); }), }; }