Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

only show trigger phrase settings on main and lora

Browse Source
This commit is contained in:
Mary Hipp 2024-03-04 12:08:23 -05:00 committed by psychedelicious
parent 2db5eaf907
commit caafbf2f0d
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelPanel/Metadata/ModelMetadata.tsx
View File

@ -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 (
<Flex flexDir="column" height="full" gap="3">
<Box layerStyle="second" borderRadius="base" p={3}>
<TriggerPhrases />
</Box>
{shouldShowTriggerPhraseSettings && (
<Box layerStyle="second" borderRadius="base" p={3}>
<TriggerPhrases />
</Box>
)}
<DataViewer label="metadata" data={data?.source_api_response || {}} />
</Flex>
);