github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-13 16:45:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): ref image model types

Browse Source
This commit is contained in:
psychedelicious
2025-07-04 20:23:32 +10:00
parent 76233241f0
commit 522eeda2e2
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/frontend/web/src/features/controlLayers/components/RefImage/RefImageModel.tsx
View File

@@ -5,11 +5,12 @@ import { selectBase } from 'features/controlLayers/store/paramsSlice';
import { memo, useCallback, useMemo } from 'react';
import { useTranslation } from 'react-i18next';
import { useGlobalReferenceImageModels } from 'services/api/hooks/modelsByType';
import type { AnyModelConfig, ApiModelConfig, FLUXReduxModelConfig, IPAdapterModelConfig } from 'services/api/types';
type RefImageModelConfig = ReturnType<typeof useGlobalReferenceImageModels>[0][number];
type Props = {
modelKey: string | null;
onChangeModel: (modelConfig: IPAdapterModelConfig | FLUXReduxModelConfig | ApiModelConfig) => void;
onChangeModel: (modelConfig: RefImageModelConfig) => void;
};
export const RefImageModel = memo(({ modelKey, onChangeModel }: Props) => {
@@ -19,7 +20,7 @@ export const RefImageModel = memo(({ modelKey, onChangeModel }: Props) => {
const selectedModel = useMemo(() => modelConfigs.find((m) => m.key === modelKey), [modelConfigs, modelKey]);
const _onChangeModel = useCallback(
(modelConfig: IPAdapterModelConfig | FLUXReduxModelConfig | ApiModelConfig | null) => {
(modelConfig: RefImageModelConfig | null) => {
if (!modelConfig) {
return;
}
@@ -29,7 +30,7 @@ export const RefImageModel = memo(({ modelKey, onChangeModel }: Props) => {
);
const getIsDisabled = useCallback(
(model: AnyModelConfig): boolean => {
(model: RefImageModelConfig): boolean => {
const hasMainModel = Boolean(currentBaseModel);
const hasSameBase = currentBaseModel === model.base;
return !hasMainModel || !hasSameBase;