Skip to content

Commit 522eeda

Browse files
fix(ui): ref image model types
1 parent 7623324 commit 522eeda

File tree

1 file changed

+5
-4
lines changed
  • invokeai/frontend/web/src/features/controlLayers/components/RefImage

1 file changed

+5
-4
lines changed

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

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,12 @@ import { selectBase } from 'features/controlLayers/store/paramsSlice';
55
import { memo, useCallback, useMemo } from 'react';
66
import { useTranslation } from 'react-i18next';
77
import { useGlobalReferenceImageModels } from 'services/api/hooks/modelsByType';
8-
import type { AnyModelConfig, ApiModelConfig, FLUXReduxModelConfig, IPAdapterModelConfig } from 'services/api/types';
8+
9+
type RefImageModelConfig = ReturnType<typeof useGlobalReferenceImageModels>[0][number];
910

1011
type Props = {
1112
modelKey: string | null;
12-
onChangeModel: (modelConfig: IPAdapterModelConfig | FLUXReduxModelConfig | ApiModelConfig) => void;
13+
onChangeModel: (modelConfig: RefImageModelConfig) => void;
1314
};
1415

1516
export const RefImageModel = memo(({ modelKey, onChangeModel }: Props) => {
@@ -19,7 +20,7 @@ export const RefImageModel = memo(({ modelKey, onChangeModel }: Props) => {
1920
const selectedModel = useMemo(() => modelConfigs.find((m) => m.key === modelKey), [modelConfigs, modelKey]);
2021

2122
const _onChangeModel = useCallback(
22-
(modelConfig: IPAdapterModelConfig | FLUXReduxModelConfig | ApiModelConfig | null) => {
23+
(modelConfig: RefImageModelConfig | null) => {
2324
if (!modelConfig) {
2425
return;
2526
}
@@ -29,7 +30,7 @@ export const RefImageModel = memo(({ modelKey, onChangeModel }: Props) => {
2930
);
3031

3132
const getIsDisabled = useCallback(
32-
(model: AnyModelConfig): boolean => {
33+
(model: RefImageModelConfig): boolean => {
3334
const hasMainModel = Boolean(currentBaseModel);
3435
const hasSameBase = currentBaseModel === model.base;
3536
return !hasMainModel || !hasSameBase;

0 commit comments

Comments
 (0)