github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update ui ModelFormatBadge to support GGUF.

Browse Source
This commit is contained in:
Ryan Dick
2024-10-02 21:19:56 +00:00
committed by Kent Keirsey
parent 3d4bd71098
commit c29259ccdb
2 changed files with 105 additions and 13 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

@@ -16,6 +16,7 @@ const FORMAT_NAME_MAP: Record<AnyModelConfig['format'], string> = {
t5_encoder: 't5_encoder',
bnb_quantized_int8b: 'bnb_quantized_int8b',
bnb_quantized_nf4b: 'quantized',
gguf_quantized: 'gguf',
};
const FORMAT_COLOR_MAP: Record<AnyModelConfig['format'], string> = {
@@ -28,6 +29,7 @@ const FORMAT_COLOR_MAP: Record<AnyModelConfig['format'], string> = {
t5_encoder: 'base',
bnb_quantized_int8b: 'base',
bnb_quantized_nf4b: 'base',
gguf_quantized: 'base',
};
const ModelFormatBadge = ({ format }: Props) => {