From b88f4a24d0926eab9cee1df2b0ad19085bc1a153 Mon Sep 17 00:00:00 2001 From: Billy Date: Mon, 23 Jun 2025 14:01:41 +1000 Subject: [PATCH] Frontend types --- .../subpanels/ModelManagerPanel/ModelFormatBadge.tsx | 2 ++ invokeai/frontend/web/src/services/api/types.ts | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx index 18289c350f..2dfb691008 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx @@ -18,10 +18,12 @@ const FORMAT_NAME_MAP: Record = { bnb_quantized_nf4b: 'quantized', gguf_quantized: 'gguf', api: 'api', + omi: 'omi', }; const FORMAT_COLOR_MAP: Record = { diffusers: 'base', + omi: 'base', lycoris: 'base', checkpoint: 'orange', invokeai: 'base', diff --git a/invokeai/frontend/web/src/services/api/types.ts b/invokeai/frontend/web/src/services/api/types.ts index 14882f43e5..a445a35c2c 100644 --- a/invokeai/frontend/web/src/services/api/types.ts +++ b/invokeai/frontend/web/src/services/api/types.ts @@ -46,7 +46,7 @@ export type BaseModelType = S['BaseModelType']; export type ControlLoRAModelConfig = S['ControlLoRALyCORISConfig'] | S['ControlLoRADiffusersConfig']; // TODO(MM2): Can we make key required in the pydantic model? -export type LoRAModelConfig = S['LoRADiffusersConfig'] | S['LoRALyCORISConfig']; +export type LoRAModelConfig = S['LoRADiffusersConfig'] | S['LoRALyCORISConfig'] | S['LoRAOmiConfig']; // TODO(MM2): Can we rename this from Vae -> VAE export type VAEModelConfig = S['VAECheckpointConfig'] | S['VAEDiffusersConfig']; export type ControlNetModelConfig = S['ControlNetDiffusersConfig'] | S['ControlNetCheckpointConfig'];