diff --git a/invokeai/frontend/web/src/features/nodes/components/fields/RefinerModelInputFieldComponent.tsx b/invokeai/frontend/web/src/features/nodes/components/fields/RefinerModelInputFieldComponent.tsx index 2a7531b59d..28c6567e8d 100644 --- a/invokeai/frontend/web/src/features/nodes/components/fields/RefinerModelInputFieldComponent.tsx +++ b/invokeai/frontend/web/src/features/nodes/components/fields/RefinerModelInputFieldComponent.tsx @@ -16,6 +16,7 @@ import { useTranslation } from 'react-i18next'; import { REFINER_BASE_MODELS } from 'services/api/constants'; import { useGetMainModelsQuery } from 'services/api/endpoints/models'; import { FieldComponentProps } from './types'; +import { useFeatureStatus } from 'features/system/hooks/useFeatureStatus'; const RefinerModelInputFieldComponent = ( props: FieldComponentProps< @@ -27,7 +28,7 @@ const RefinerModelInputFieldComponent = ( const dispatch = useAppDispatch(); const { t } = useTranslation(); - + const isSyncModelEnabled = useFeatureStatus('syncModels').isFeatureEnabled; const { data: refinerModels, isLoading } = useGetMainModelsQuery(REFINER_BASE_MODELS); @@ -107,9 +108,11 @@ const RefinerModelInputFieldComponent = ( disabled={data.length === 0} onChange={handleChangeModel} /> - - - + {isSyncModelEnabled && ( + + + + )} ); }; diff --git a/invokeai/frontend/web/src/features/sdxl/components/SDXLRefiner/ParamSDXLRefinerModelSelect.tsx b/invokeai/frontend/web/src/features/sdxl/components/SDXLRefiner/ParamSDXLRefinerModelSelect.tsx index 4984e74964..cae40bbff3 100644 --- a/invokeai/frontend/web/src/features/sdxl/components/SDXLRefiner/ParamSDXLRefinerModelSelect.tsx +++ b/invokeai/frontend/web/src/features/sdxl/components/SDXLRefiner/ParamSDXLRefinerModelSelect.tsx @@ -8,6 +8,7 @@ import IAIMantineSearchableSelect from 'common/components/IAIMantineSearchableSe import { MODEL_TYPE_MAP } from 'features/parameters/types/constants'; import { modelIdToMainModelParam } from 'features/parameters/util/modelIdToMainModelParam'; import { refinerModelChanged } from 'features/sdxl/store/sdxlSlice'; +import { useFeatureStatus } from 'features/system/hooks/useFeatureStatus'; import SyncModelsButton from 'features/ui/components/tabs/ModelManager/subpanels/ModelManagerSettingsPanel/SyncModelsButton'; import { forEach } from 'lodash-es'; import { memo, useCallback, useMemo } from 'react'; @@ -22,6 +23,7 @@ const selector = createSelector( const ParamSDXLRefinerModelSelect = () => { const dispatch = useAppDispatch(); + const isSyncModelEnabled = useFeatureStatus('syncModels').isFeatureEnabled; const { model } = useAppSelector(selector); @@ -97,9 +99,11 @@ const ParamSDXLRefinerModelSelect = () => { onChange={handleChangeModel} w="100%" /> - - - + {isSyncModelEnabled && ( + + + + )} ); };