Skip to content

Commit db15c54

Browse files
author
Attila Cseh
committed
selectors refined
1 parent df4754a commit db15c54

30 files changed

+142
-144
lines changed

invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/appConfigReceived.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import type { AppStartListening } from 'app/store/store';
2-
import { selectActiveParams, setInfillMethod } from 'features/controlLayers/store/paramsSlice';
2+
import { selectActiveTabParams, setInfillMethod } from 'features/controlLayers/store/paramsSlice';
33
import { shouldUseNSFWCheckerChanged, shouldUseWatermarkerChanged } from 'features/system/store/systemSlice';
44
import { appInfoApi } from 'services/api/endpoints/appInfo';
55

@@ -9,7 +9,7 @@ export const addAppConfigReceivedListener = (startAppListening: AppStartListenin
99
effect: (action, api) => {
1010
const { getState, dispatch } = api;
1111
const { infill_methods = [], nsfw_methods = [], watermarking_methods = [] } = action.payload;
12-
const infillMethod = selectActiveParams(getState()).infillMethod;
12+
const infillMethod = selectActiveTabParams(getState()).infillMethod;
1313

1414
if (!infill_methods.includes(infillMethod)) {
1515
// If the selected infill method does not exist, prefer 'lama' if it's in the list, otherwise 'tile'.

invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/modelSelected.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import {
77
selectActiveCanvasStagingAreaSessionId,
88
} from 'features/controlLayers/store/canvasStagingAreaSlice';
99
import { loraIsEnabledChanged } from 'features/controlLayers/store/lorasSlice';
10-
import { selectActiveParams, syncedToOptimalDimension, vaeSelected } from 'features/controlLayers/store/paramsSlice';
10+
import { selectActiveTabParams, syncedToOptimalDimension, vaeSelected } from 'features/controlLayers/store/paramsSlice';
1111
import { refImageModelChanged, selectReferenceImageEntities } from 'features/controlLayers/store/refImagesSlice';
1212
import {
1313
selectActiveCanvas,
@@ -47,7 +47,7 @@ export const addModelSelectedListener = (startAppListening: AppStartListening) =
4747

4848
const newModel = result.data;
4949
const newBase = newModel.base;
50-
const params = selectActiveParams(state);
50+
const params = selectActiveTabParams(state);
5151
const didBaseModelChange = params.model?.base !== newBase;
5252

5353
if (didBaseModelChange) {

invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/modelsLoaded.ts

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import {
77
clipEmbedModelSelected,
88
fluxVAESelected,
99
refinerModelChanged,
10-
selectActiveParams,
10+
selectActiveTabParams,
1111
t5EncoderModelSelected,
1212
vaeSelected,
1313
} from 'features/controlLayers/store/paramsSlice';
@@ -104,7 +104,7 @@ type ModelHandler = (
104104
) => undefined;
105105

106106
const handleMainModels: ModelHandler = (models, state, dispatch, log) => {
107-
const selectedMainModel = selectActiveParams(state).model;
107+
const selectedMainModel = selectActiveTabParams(state).model;
108108
const allMainModels = models.filter(isNonRefinerMainModelConfig).sort((a) => (a.base === 'sdxl' ? -1 : 1));
109109

110110
const firstModel = allMainModels[0];
@@ -145,7 +145,7 @@ const handleMainModels: ModelHandler = (models, state, dispatch, log) => {
145145
};
146146

147147
const handleRefinerModels: ModelHandler = (models, state, dispatch, log) => {
148-
const selectedRefinerModel = selectActiveParams(state).refinerModel;
148+
const selectedRefinerModel = selectActiveTabParams(state).refinerModel;
149149

150150
// `null` is a valid refiner model - no need to do anything.
151151
if (selectedRefinerModel === null) {
@@ -169,7 +169,7 @@ const handleRefinerModels: ModelHandler = (models, state, dispatch, log) => {
169169
};
170170

171171
const handleVAEModels: ModelHandler = (models, state, dispatch, log) => {
172-
const selectedVAEModel = selectActiveParams(state).vae;
172+
const selectedVAEModel = selectActiveTabParams(state).vae;
173173

174174
// `null` is a valid VAE - it means "use the VAE baked into the currently-selected main model"
175175
if (selectedVAEModel === null) {
@@ -418,7 +418,7 @@ const handleTileControlNetModel: ModelHandler = (models, state, dispatch, log) =
418418
};
419419

420420
const handleT5EncoderModels: ModelHandler = (models, state, dispatch, log) => {
421-
const selectedT5EncoderModel = selectActiveParams(state).t5EncoderModel;
421+
const selectedT5EncoderModel = selectActiveTabParams(state).t5EncoderModel;
422422
const t5EncoderModels = models.filter((m) => isT5EncoderModelConfig(m));
423423

424424
// If the currently selected model is available, we don't need to do anything
@@ -446,7 +446,7 @@ const handleT5EncoderModels: ModelHandler = (models, state, dispatch, log) => {
446446
};
447447

448448
const handleCLIPEmbedModels: ModelHandler = (models, state, dispatch, log) => {
449-
const selectedCLIPEmbedModel = selectActiveParams(state).clipEmbedModel;
449+
const selectedCLIPEmbedModel = selectActiveTabParams(state).clipEmbedModel;
450450
const CLIPEmbedModels = models.filter((m) => isCLIPEmbedModelConfig(m));
451451

452452
// If the currently selected model is available, we don't need to do anything
@@ -474,7 +474,7 @@ const handleCLIPEmbedModels: ModelHandler = (models, state, dispatch, log) => {
474474
};
475475

476476
const handleFLUXVAEModels: ModelHandler = (models, state, dispatch, log) => {
477-
const selectedFLUXVAEModel = selectActiveParams(state).fluxVAE;
477+
const selectedFLUXVAEModel = selectActiveTabParams(state).fluxVAE;
478478
const fluxVAEModels = models.filter((m) => isFluxVAEModelConfig(m));
479479

480480
// If the currently selected model is available, we don't need to do anything

invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/setDefaultSettings.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import {
77
} from 'features/controlLayers/store/canvasStagingAreaSlice';
88
import {
99
heightChanged,
10-
selectActiveParams,
10+
selectActiveTabParams,
1111
setCfgRescaleMultiplier,
1212
setCfgScale,
1313
setGuidance,
@@ -42,7 +42,7 @@ export const addSetDefaultSettingsListener = (startAppListening: AppStartListeni
4242
const { dispatch, getState } = api;
4343
const state = getState();
4444

45-
const currentModel = selectActiveParams(state).model;
45+
const currentModel = selectActiveTabParams(state).model;
4646

4747
if (!currentModel) {
4848
return;

invokeai/frontend/web/src/common/hooks/useGroupedModelCombobox.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ import { createSelector } from '@reduxjs/toolkit';
33
import { useAppSelector } from 'app/store/storeHooks';
44
import type { GroupBase } from 'chakra-react-select';
55
import { groupBy, reduce } from 'es-toolkit/compat';
6-
import { selectActiveParams } from 'features/controlLayers/store/paramsSlice';
6+
import { selectActiveTabParams } from 'features/controlLayers/store/paramsSlice';
77
import type { ModelIdentifierField } from 'features/nodes/types/common';
88
import { selectSystemShouldEnableModelDescriptions } from 'features/system/store/systemSlice';
99
import { useCallback, useMemo } from 'react';
@@ -31,7 +31,7 @@ const groupByBaseFunc = <T extends AnyModelConfig>(model: T) => model.base.toUpp
3131
const groupByBaseAndTypeFunc = <T extends AnyModelConfig>(model: T) =>
3232
`${model.base.toUpperCase()} / ${model.type.replaceAll('_', ' ').toUpperCase()}`;
3333

34-
const selectBaseWithSDXLFallback = createSelector(selectActiveParams, (params) => params.model?.base ?? 'sdxl');
34+
const selectBaseWithSDXLFallback = createSelector(selectActiveTabParams, (params) => params.model?.base ?? 'sdxl');
3535

3636
export const useGroupedModelCombobox = <T extends AnyModelConfig>(
3737
arg: UseGroupedModelComboboxArg<T>

0 commit comments

Comments
 (0)