only show trigger phrase settings on main and lora

This commit is contained in:
Mary Hipp 2024-03-04 12:08:23 -05:00 committed by psychedelicious
parent 2db5eaf907
commit caafbf2f0d

View File

@ -2,19 +2,32 @@ import { Box, Flex } from '@invoke-ai/ui-library';
import { skipToken } from '@reduxjs/toolkit/query'; import { skipToken } from '@reduxjs/toolkit/query';
import { useAppSelector } from 'app/store/storeHooks'; import { useAppSelector } from 'app/store/storeHooks';
import DataViewer from 'features/gallery/components/ImageMetadataViewer/DataViewer'; import DataViewer from 'features/gallery/components/ImageMetadataViewer/DataViewer';
import { useMemo } from 'react';
import { useGetModelConfigQuery } from 'services/api/endpoints/models'; import { useGetModelConfigQuery } from 'services/api/endpoints/models';
import type { ModelType } from 'services/api/types';
import { TriggerPhrases } from './TriggerPhrases'; import { TriggerPhrases } from './TriggerPhrases';
const MODEL_TYPE_TRIGGER_PHRASE: ModelType[] = ['main', 'lora'];
export const ModelMetadata = () => { export const ModelMetadata = () => {
const selectedModelKey = useAppSelector((s) => s.modelmanagerV2.selectedModelKey); const selectedModelKey = useAppSelector((s) => s.modelmanagerV2.selectedModelKey);
const { data } = useGetModelConfigQuery(selectedModelKey ?? skipToken); const { data } = useGetModelConfigQuery(selectedModelKey ?? skipToken);
const shouldShowTriggerPhraseSettings = useMemo(() => {
if (!data?.type) {
return false;
}
return MODEL_TYPE_TRIGGER_PHRASE.includes(data.type);
}, [data]);
return ( return (
<Flex flexDir="column" height="full" gap="3"> <Flex flexDir="column" height="full" gap="3">
<Box layerStyle="second" borderRadius="base" p={3}> {shouldShowTriggerPhraseSettings && (
<TriggerPhrases /> <Box layerStyle="second" borderRadius="base" p={3}>
</Box> <TriggerPhrases />
</Box>
)}
<DataViewer label="metadata" data={data?.source_api_response || {}} /> <DataViewer label="metadata" data={data?.source_api_response || {}} />
</Flex> </Flex>
); );