fix: Model type is not match comfyui

src/utils/model.ts

@@ -1,47 +1,25 @@
 const loader = {
   checkpoints: 'CheckpointLoaderSimple',
+  loras: 'LoraLoader',
+  vae: 'VAELoader',
   clip: 'CLIPLoader',
+  diffusion_models: 'UNETLoader',
+  unet: 'UNETLoader',
   clip_vision: 'CLIPVisionLoader',
-  controlnet: 'ControlNetLoader',
+  style_models: 'StyleModelLoader',
+  embeddings: undefined,
   diffusers: 'DiffusersLoader',
-  diffusion_models: 'DiffusersLoader',
-  embeddings: 'Embedding',
+  vae_approx: undefined,
+  controlnet: 'ControlNetLoader',
   gligen: 'GLIGENLoader',
+  upscale_models: 'UpscaleModelLoader',
   hypernetworks: 'HypernetworkLoader',
   photomaker: 'PhotoMakerLoader',
-  loras: 'LoraLoader',
-  style_models: 'StyleModelLoader',
-  unet: 'UNETLoader',
-  upscale_models: 'UpscaleModelLoader',
-  vae: 'VAELoader',
-  vae_approx: undefined,
+  classifiers: undefined,
 }
 
-const display = {
-  all: 'ALL',
-  checkpoints: 'Checkpoint',
-  clip: 'Clip',
-  clip_vision: 'Clip Vision',
-  controlnet: 'Controlnet',
-  diffusers: 'Diffusers',
-  diffusion_models: 'Diffusers',
-  embeddings: 'embedding',
-  gligen: 'Gligen',
-  hypernetworks: 'Hypernetwork',
-  photomaker: 'Photomaker',
-  loras: 'LoRA',
-  style_models: 'Style Model',
-  unet: 'Unet',
-  upscale_models: 'Upscale Model',
-  vae: 'VAE',
-  vae_approx: 'VAE approx',
-}
-
-export const resolveModelType = (type: string) => {
-  return {
-    display: display[type],
-    loader: loader[type],
-  }
+export const resolveModelTypeLoader = (type: string) => {
+  return loader[type]
 }
 
 export const genModelKey = (model: BaseModel) => {