@@ -5,11 +5,12 @@ import { selectBase } from 'features/controlLayers/store/paramsSlice';
55import { memo , useCallback , useMemo } from 'react' ;
66import { useTranslation } from 'react-i18next' ;
77import { 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
1011type Props = {
1112 modelKey : string | null ;
12- onChangeModel : ( modelConfig : IPAdapterModelConfig | FLUXReduxModelConfig | ApiModelConfig ) => void ;
13+ onChangeModel : ( modelConfig : RefImageModelConfig ) => void ;
1314} ;
1415
1516export 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