From dc4c63e6541cb8882dd2c0666c6dac51b185530b Mon Sep 17 00:00:00 2001
From: BennyKok <itechbenny@gmail.com>
Date: Sat, 13 Jan 2024 23:45:59 +0800
Subject: [PATCH] fix: model type

---
 web/src/components/custom-form/ModelPickerView.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/web/src/components/custom-form/ModelPickerView.tsx b/web/src/components/custom-form/ModelPickerView.tsx
index 10b3565..12ed3eb 100644
--- a/web/src/components/custom-form/ModelPickerView.tsx
+++ b/web/src/components/custom-form/ModelPickerView.tsx
@@ -181,7 +181,7 @@ function mapModelsList(
       const v = item.modelVersions[0];
       return {
         name: `${item.name} ${v.name} (${v.files[0].name})`,
-        type: v.files[0].type.toLowerCase(),
+        type: item.type.toLowerCase(),
         base: v.baseModel,
         save_path: "default",
         description: item.description,