diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelPanel/Metadata/ModelMetadata.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelPanel/Metadata/ModelMetadata.tsx index 7ffeadb510..2c6294a987 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelPanel/Metadata/ModelMetadata.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelPanel/Metadata/ModelMetadata.tsx @@ -2,19 +2,32 @@ import { Box, Flex } from '@invoke-ai/ui-library'; import { skipToken } from '@reduxjs/toolkit/query'; import { useAppSelector } from 'app/store/storeHooks'; import DataViewer from 'features/gallery/components/ImageMetadataViewer/DataViewer'; +import { useMemo } from 'react'; import { useGetModelConfigQuery } from 'services/api/endpoints/models'; +import type { ModelType } from 'services/api/types'; import { TriggerPhrases } from './TriggerPhrases'; +const MODEL_TYPE_TRIGGER_PHRASE: ModelType[] = ['main', 'lora']; + export const ModelMetadata = () => { const selectedModelKey = useAppSelector((s) => s.modelmanagerV2.selectedModelKey); const { data } = useGetModelConfigQuery(selectedModelKey ?? skipToken); + const shouldShowTriggerPhraseSettings = useMemo(() => { + if (!data?.type) { + return false; + } + return MODEL_TYPE_TRIGGER_PHRASE.includes(data.type); + }, [data]); + return ( - - - + {shouldShowTriggerPhraseSettings && ( + + + + )} );