fix: Model type is not match comfyui

src/components/DialogManager.vue

@@ -86,7 +86,7 @@ import { chunk } from 'lodash'
 import { defineResizeCallback } from 'hooks/resize'
 import { genModelKey } from 'utils/model'
 
-const { isMobile, cardWidth, gutter, aspect } = useConfig()
+const { isMobile, cardWidth, gutter, aspect, modelFolders } = useConfig()
 
 const { data } = useModels()
 const { t } = useI18n()
@@ -94,32 +94,17 @@ const { t } = useI18n()
 const searchContent = ref<string>()
 
 const currentType = ref('all')
-const typeOptions = ref(
-  [
-    { label: 'ALL', value: 'all' },
-    { label: 'Checkpoint', value: 'checkpoints' },
-    { label: 'embedding', value: 'embeddings' },
-    { label: 'Hypernetwork', value: 'hypernetworks' },
-    { label: 'Lora', value: 'loras' },
-    { label: 'VAE', value: 'vae' },
-    { label: 'VAE approx', value: 'vae_approx' },
-    { label: 'Controlnet', value: 'controlnet' },
-    { label: 'Clip', value: 'clip' },
-    { label: 'Clip Vision', value: 'clip_vision' },
-    { label: 'Diffusers', value: 'diffusers' },
-    { label: 'Gligen', value: 'gligen' },
-    { label: 'Photomaker', value: 'photomaker' },
-    { label: 'Style Models', value: 'style_models' },
-    { label: 'Unet', value: 'unet' },
-  ].map((item) => {
+const typeOptions = computed(() => {
+  return ['all', ...Object.keys(modelFolders.value)].map((type) => {
     return {
-      ...item,
+      label: type,
+      value: type,
       command: () => {
-        currentType.value = item.value
+        currentType.value = type
       },
     }
-  }),
-)
+  })
+})
 
 const sortOrder = ref('name')
 const sortOrderOptions = ref(

src/components/ModelBaseInfo.vue

@@ -45,7 +45,6 @@ import ResponseInput from 'components/ResponseInput.vue'
 import ResponseSelect from 'components/ResponseSelect.vue'
 import { useConfig } from 'hooks/config'
 import { useModelBaseInfo } from 'hooks/model'
-import { resolveModelType } from 'utils/model'
 import { computed } from 'vue'
 
 const editable = defineModel<boolean>('editable')
@@ -58,7 +57,7 @@ const typeOptions = computed(() => {
   return Object.keys(modelFolders.value).map((curr) => {
     return {
       value: curr,
-      label: resolveModelType(curr).display,
+      label: curr,
       command: () => {
         type.value = curr
         pathIndex.value = 0

src/components/ModelCard.vue

@@ -30,7 +30,7 @@
           <div class="flex flex-row items-start justify-between">
             <div class="flex items-center rounded-full bg-black/30 px-3 py-2">
               <div class="font-bold @lg:text-xs">
-                {{ displayType }}
+                {{ model.type }}
               </div>
             </div>
 
@@ -67,7 +67,7 @@
 <script setup lang="ts">
 import DialogModelDetail from 'components/DialogModelDetail.vue'
 import Button from 'primevue/button'
-import { genModelKey, resolveModelType } from 'utils/model'
+import { genModelKey } from 'utils/model'
 import { computed } from 'vue'
 import { useModelNodeAction } from 'hooks/model'
 import { useDialog } from 'hooks/dialog'
@@ -92,7 +92,6 @@ const openDetailDialog = () => {
   })
 }
 
-const displayType = computed(() => resolveModelType(props.model.type).display)
 const preview = computed(() =>
   Array.isArray(props.model.preview)
     ? props.model.preview[0]