diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx index a3c9c82d0e..4fc8390890 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx @@ -12,10 +12,18 @@ type Props = { export const StarterModelsResultItem = memo(({ result }: Props) => { const { t } = useTranslation(); const allSources = useMemo(() => { - const _allSources = [{ source: result.source, config: { name: result.name, description: result.description } }]; + const _allSources = [ + { + source: result.source, + config: { name: result.name, description: result.description, type: result.type, base: result.base }, + }, + ]; if (result.dependencies) { for (const d of result.dependencies) { - _allSources.push({ source: d.source, config: { name: d.name, description: d.description } }); + _allSources.push({ + source: d.source, + config: { name: d.name, description: d.description, type: d.type, base: d.base }, + }); } } return _allSources; 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 2cf4e25354..3eb0a91d67 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelBaseBadge.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelBaseBadge.tsx @@ -13,7 +13,7 @@ const BASE_COLOR_MAP: Record = { 'sd-2': 'teal', sdxl: 'invokeBlue', 'sdxl-refiner': 'invokeBlue', - flux: 'invokeBlue', + flux: 'gold', }; const ModelBaseBadge = ({ base }: Props) => { 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 a4690662c3..167588ddd3 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelFormatBadge.tsx @@ -13,6 +13,9 @@ const FORMAT_NAME_MAP: Record = { invokeai: 'internal', embedding_file: 'embedding', embedding_folder: 'embedding', + t5_encoder: 't5_encoder', + t5_encoder_8b: 't5_encoder_8b', + bnb_quantized_nf4b: 'quantized', }; const FORMAT_COLOR_MAP: Record = { @@ -22,6 +25,9 @@ const FORMAT_COLOR_MAP: Record = { invokeai: 'base', embedding_file: 'base', embedding_folder: 'base', + t5_encoder: 'base', + t5_encoder_8b: 'base', + bnb_quantized_nf4b: 'base', }; const ModelFormatBadge = ({ format }: Props) => {