Revert "Revert "feat(ui): remove special handling for t2i vs controlnet""

This reverts commit bdf4c4944c.

invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/index.ts

@@ -12,6 +12,7 @@ import { addFirstListImagesListener } from './listeners/addFirstListImagesListen
 import { addAnyEnqueuedListener } from './listeners/anyEnqueued';
 import { addAppConfigReceivedListener } from './listeners/appConfigReceived';
 import { addAppStartedListener } from './listeners/appStarted';
+import { addBatchEnqueuedListener } from './listeners/batchEnqueued';
 import { addDeleteBoardAndImagesFulfilledListener } from './listeners/boardAndImagesDeleted';
 import { addBoardIdSelectedListener } from './listeners/boardIdSelected';
 import { addCanvasCopiedToClipboardListener } from './listeners/canvasCopiedToClipboard';
@@ -71,8 +72,6 @@ import { addStagingAreaImageSavedListener } from './listeners/stagingAreaImageSa
 import { addTabChangedListener } from './listeners/tabChanged';
 import { addUpscaleRequestedListener } from './listeners/upscaleRequested';
 import { addWorkflowLoadedListener } from './listeners/workflowLoaded';
-import { addBatchEnqueuedListener } from './listeners/batchEnqueued';
-import { addControlAdapterAddedOrEnabledListener } from './listeners/controlAdapterAddedOrEnabled';
 
 export const listenerMiddleware = createListenerMiddleware();
 
@@ -200,7 +199,3 @@ addTabChangedListener();
 
 // Dynamic prompts
 addDynamicPromptsListener();
-
-// Display toast when controlnet or t2i adapter enabled
-// TODO: Remove when they can both be enabled at same time
-addControlAdapterAddedOrEnabledListener();

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

@@ -1,87 +0,0 @@
-import { isAnyOf } from '@reduxjs/toolkit';
-import {
-  controlAdapterAdded,
-  controlAdapterAddedFromImage,
-  controlAdapterIsEnabledChanged,
-  controlAdapterRecalled,
-  selectControlAdapterAll,
-  selectControlAdapterById,
-} from 'features/controlAdapters/store/controlAdaptersSlice';
-import { ControlAdapterType } from 'features/controlAdapters/store/types';
-import { addToast } from 'features/system/store/systemSlice';
-import i18n from 'i18n';
-import { startAppListening } from '..';
-
-const isAnyControlAdapterAddedOrEnabled = isAnyOf(
-  controlAdapterAdded,
-  controlAdapterAddedFromImage,
-  controlAdapterRecalled,
-  controlAdapterIsEnabledChanged
-);
-
-/**
- * Until we can have both controlnet and t2i adapter enabled at once, they are mutually exclusive
- * This displays a toast when one is enabled and the other is already enabled, or one is added
- * with the other enabled
- */
-export const addControlAdapterAddedOrEnabledListener = () => {
-  startAppListening({
-    matcher: isAnyControlAdapterAddedOrEnabled,
-    effect: async (action, { dispatch, getOriginalState }) => {
-      const controlAdapters = getOriginalState().controlAdapters;
-
-      const hasEnabledControlNets = selectControlAdapterAll(
-        controlAdapters
-      ).some((ca) => ca.isEnabled && ca.type === 'controlnet');
-
-      const hasEnabledT2IAdapters = selectControlAdapterAll(
-        controlAdapters
-      ).some((ca) => ca.isEnabled && ca.type === 't2i_adapter');
-
-      let caType: ControlAdapterType | null = null;
-
-      if (controlAdapterAdded.match(action)) {
-        caType = action.payload.type;
-      }
-
-      if (controlAdapterAddedFromImage.match(action)) {
-        caType = action.payload.type;
-      }
-
-      if (controlAdapterRecalled.match(action)) {
-        caType = action.payload.type;
-      }
-
-      if (controlAdapterIsEnabledChanged.match(action)) {
-        const _caType = selectControlAdapterById(
-          controlAdapters,
-          action.payload.id
-        )?.type;
-        if (!_caType) {
-          return;
-        }
-        caType = _caType;
-      }
-
-      if (
-        (caType === 'controlnet' && hasEnabledT2IAdapters) ||
-        (caType === 't2i_adapter' && hasEnabledControlNets)
-      ) {
-        const title =
-          caType === 'controlnet'
-            ? i18n.t('controlnet.controlNetEnabledT2IDisabled')
-            : i18n.t('controlnet.t2iEnabledControlNetDisabled');
-
-        const description = i18n.t('controlnet.controlNetT2IMutexDesc');
-
-        dispatch(
-          addToast({
-            title,
-            description,
-            status: 'warning',
-          })
-        );
-      }
-    },
-  });
-};

invokeai/frontend/web/src/features/controlAdapters/store/controlAdaptersSlice.ts

@@ -88,61 +88,6 @@ export const selectValidT2IAdapters = (controlAdapters: ControlAdaptersState) =>
           (ca.processorType === 'none' && Boolean(ca.controlImage)))
     );
 
-// TODO: I think we can safely remove this?
-// const disableAllIPAdapters = (
-//   state: ControlAdaptersState,
-//   exclude?: string
-// ) => {
-//   const updates: Update<ControlAdapterConfig>[] = selectAllIPAdapters(state)
-//     .filter((ca) => ca.id !== exclude)
-//     .map((ca) => ({
-//       id: ca.id,
-//       changes: { isEnabled: false },
-//     }));
-//   caAdapter.updateMany(state, updates);
-// };
-
-const disableAllControlNets = (
-  state: ControlAdaptersState,
-  exclude?: string
-) => {
-  const updates: Update<ControlAdapterConfig>[] = selectAllControlNets(state)
-    .filter((ca) => ca.id !== exclude)
-    .map((ca) => ({
-      id: ca.id,
-      changes: { isEnabled: false },
-    }));
-  caAdapter.updateMany(state, updates);
-};
-
-const disableAllT2IAdapters = (
-  state: ControlAdaptersState,
-  exclude?: string
-) => {
-  const updates: Update<ControlAdapterConfig>[] = selectAllT2IAdapters(state)
-    .filter((ca) => ca.id !== exclude)
-    .map((ca) => ({
-      id: ca.id,
-      changes: { isEnabled: false },
-    }));
-  caAdapter.updateMany(state, updates);
-};
-
-const disableIncompatibleControlAdapters = (
-  state: ControlAdaptersState,
-  type: ControlAdapterType,
-  exclude?: string
-) => {
-  if (type === 'controlnet') {
-    // we cannot do controlnet + t2i adapter, if we are enabled a controlnet, disable all t2is
-    disableAllT2IAdapters(state, exclude);
-  }
-  if (type === 't2i_adapter') {
-    // we cannot do controlnet + t2i adapter, if we are enabled a t2i, disable controlnets
-    disableAllControlNets(state, exclude);
-  }
-};
-
 export const controlAdaptersSlice = createSlice({
   name: 'controlAdapters',
   initialState: initialControlAdapterState,
@@ -158,7 +103,6 @@ export const controlAdaptersSlice = createSlice({
       ) => {
         const { id, type, overrides } = action.payload;
         caAdapter.addOne(state, buildControlAdapter(id, type, overrides));
-        disableIncompatibleControlAdapters(state, type, id);
       },
       prepare: ({
         type,
@@ -175,8 +119,6 @@ export const controlAdaptersSlice = createSlice({
       action: PayloadAction<ControlAdapterConfig>
     ) => {
       caAdapter.addOne(state, action.payload);
-      const { type, id } = action.payload;
-      disableIncompatibleControlAdapters(state, type, id);
     },
     controlAdapterDuplicated: {
       reducer: (
@@ -196,8 +138,6 @@ export const controlAdaptersSlice = createSlice({
           isEnabled: true,
         });
         caAdapter.addOne(state, newControlAdapter);
-        const { type } = newControlAdapter;
-        disableIncompatibleControlAdapters(state, type, newId);
       },
       prepare: (id: string) => {
         return { payload: { id, newId: uuidv4() } };
@@ -217,7 +157,6 @@ export const controlAdaptersSlice = createSlice({
           state,
           buildControlAdapter(id, type, { controlImage })
         );
-        disableIncompatibleControlAdapters(state, type, id);
       },
       prepare: (payload: {
         type: ControlAdapterType;
@@ -235,12 +174,6 @@ export const controlAdaptersSlice = createSlice({
     ) => {
       const { id, isEnabled } = action.payload;
       caAdapter.updateOne(state, { id, changes: { isEnabled } });
-      if (isEnabled) {
-        // we are enabling a control adapter. due to limitations in the current system, we may need to disable other adapters
-        // TODO: disable when multiple IP adapters are supported
-        const ca = selectControlAdapterById(state, id);
-        ca && disableIncompatibleControlAdapters(state, ca.type, id);
-      }
     },
     controlAdapterImageChanged: (
       state,