diff --git a/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts b/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts index fe17f45940..899e5db13d 100644 --- a/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts +++ b/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts @@ -38,7 +38,11 @@ export const gallerySlice = createSlice({ state.selection = uniqBy(action.payload, (i) => i.image_name); }, imageToCompareChanged: (state, action: PayloadAction) => { - state.imageToCompare = action.payload; + if (state.imageToCompare?.image_name === action.payload?.image_name) { + state.imageToCompare = null; + } else { + state.imageToCompare = action.payload; + } if (action.payload) { state.isImageViewerOpen = true; }