diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelBaseBadge.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelBaseBadge.tsx index 7dc51d751b..e4db0daaf8 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelBaseBadge.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelBaseBadge.tsx @@ -7,9 +7,17 @@ type Props = { base: BaseModelType; }; +const BASE_COLOR_MAP: Record = { + any: 'base', + 'sd-1': 'green', + 'sd-2': 'teal', + sdxl: 'invokeBlue', + 'sdxl-refiner': 'invokeBlue', +}; + const ModelBaseBadge = ({ base }: Props) => { return ( - + {MODEL_TYPE_SHORT_MAP[base]} ); 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 5347609cb1..a4690662c3 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx @@ -6,7 +6,7 @@ type Props = { format: AnyModelConfig['format']; }; -const MODEL_FORMAT_NAME_MAP = { +const FORMAT_NAME_MAP: Record = { diffusers: 'diffusers', lycoris: 'lycoris', checkpoint: 'checkpoint', @@ -15,10 +15,19 @@ const MODEL_FORMAT_NAME_MAP = { embedding_folder: 'embedding', }; +const FORMAT_COLOR_MAP: Record = { + diffusers: 'base', + lycoris: 'base', + checkpoint: 'orange', + invokeai: 'base', + embedding_file: 'base', + embedding_folder: 'base', +}; + const ModelFormatBadge = ({ format }: Props) => { return ( - - {MODEL_FORMAT_NAME_MAP[format]} + + {FORMAT_NAME_MAP[format]} ); };