github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-12 15:24:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Frontend types

Browse Source
This commit is contained in:
Billy
2025-06-23 14:01:41 +10:00
parent 8c632f0d32
commit b88f4a24d0
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx
View File

@@ -18,10 +18,12 @@ const FORMAT_NAME_MAP: Record<AnyModelConfig['format'], string> = {
bnb_quantized_nf4b: 'quantized',
gguf_quantized: 'gguf',
api: 'api',
omi: 'omi',
};
const FORMAT_COLOR_MAP: Record<AnyModelConfig['format'], string> = {
diffusers: 'base',
omi: 'base',
lycoris: 'base',
checkpoint: 'orange',
invokeai: 'base',

2
invokeai/frontend/web/src/services/api/types.ts
View File

@@ -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'];