From 6c49bff09581122c80fb451f9223ec25eb441b7c Mon Sep 17 00:00:00 2001
From: BennyKok <itechbenny@gmail.com>
Date: Sun, 14 Jan 2024 00:15:26 +0800
Subject: [PATCH] fix: type

---
 web/src/components/custom-form/ModelPickerView.tsx | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/web/src/components/custom-form/ModelPickerView.tsx b/web/src/components/custom-form/ModelPickerView.tsx
index 12ed3eb..92c8055 100644
--- a/web/src/components/custom-form/ModelPickerView.tsx
+++ b/web/src/components/custom-form/ModelPickerView.tsx
@@ -173,6 +173,14 @@ export function ModelPickerView({
   );
 }
 
+function mapType(type: string) {
+  switch (type) {
+    case "checkpoint":
+      return "checkpoints";
+  }
+  return type;
+}
+
 function mapModelsList(
   models: z.infer<typeof CivitalModelSchema>
 ): z.infer<typeof ModelListWrapper> {
@@ -181,7 +189,7 @@ function mapModelsList(
       const v = item.modelVersions[0];
       return {
         name: `${item.name} ${v.name} (${v.files[0].name})`,
-        type: item.type.toLowerCase(),
+        type: mapType(item.type.toLowerCase()),
         base: v.baseModel,
         save_path: "default",
         description: item.description,