mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
only show trigger phrase settings on main and lora
This commit is contained in:
parent
2db5eaf907
commit
caafbf2f0d
@ -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>
|
||||||
);
|
);
|
||||||
|
Loading…
Reference in New Issue
Block a user