mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-13 15:04:59 -05:00
42 lines
1.6 KiB
TypeScript
42 lines
1.6 KiB
TypeScript
import { Combobox, FormControl, FormLabel } from '@invoke-ai/ui-library';
|
|
import { useAppDispatch, useAppSelector } from 'app/store/storeHooks';
|
|
import { useModelCombobox } from 'common/hooks/useModelCombobox';
|
|
import { clipEmbedModelSelected, selectCLIPEmbedModel } from 'features/controlLayers/store/paramsSlice';
|
|
import { zModelIdentifierField } from 'features/nodes/types/common';
|
|
import { memo, useCallback } from 'react';
|
|
import { useTranslation } from 'react-i18next';
|
|
import { useCLIPEmbedModels } from 'services/api/hooks/modelsByType';
|
|
import type { CLIPEmbedModelConfig } from 'services/api/types';
|
|
|
|
const ParamCLIPEmbedModelSelect = () => {
|
|
const dispatch = useAppDispatch();
|
|
const { t } = useTranslation();
|
|
const clipEmbedModel = useAppSelector(selectCLIPEmbedModel);
|
|
const [modelConfigs, { isLoading }] = useCLIPEmbedModels();
|
|
|
|
const _onChange = useCallback(
|
|
(clipEmbedModel: CLIPEmbedModelConfig | null) => {
|
|
if (clipEmbedModel) {
|
|
dispatch(clipEmbedModelSelected(zModelIdentifierField.parse(clipEmbedModel)));
|
|
}
|
|
},
|
|
[dispatch]
|
|
);
|
|
|
|
const { options, value, onChange, noOptionsMessage } = useModelCombobox({
|
|
modelConfigs,
|
|
onChange: _onChange,
|
|
selectedModel: clipEmbedModel,
|
|
isLoading,
|
|
});
|
|
|
|
return (
|
|
<FormControl isDisabled={!options.length} isInvalid={!options.length} minW={0} flexGrow={1}>
|
|
<FormLabel m={0}>{t('modelManager.clipEmbed')}</FormLabel>
|
|
<Combobox value={value} options={options} onChange={onChange} noOptionsMessage={noOptionsMessage} />
|
|
</FormControl>
|
|
);
|
|
};
|
|
|
|
export default memo(ParamCLIPEmbedModelSelect);
|