diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx
index e610ca0077..dfc131c87c 100644
--- a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx
+++ b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx
@@ -9,7 +9,13 @@ import CurrentImageButtons from './CurrentImageButtons';
import { ViewerToggleMenu } from './ViewerToggleMenu';
export const ViewerToolbar = memo(() => {
- const tab = useAppSelector(activeTabNameSelector);
+ const showToggle = useAppSelector((s) => {
+ const tab = activeTabNameSelector(s);
+ if (tab === 'upscaling' || tab === 'workflows') {
+ return false;
+ }
+ return true;
+ });
return (
@@ -23,7 +29,7 @@ export const ViewerToolbar = memo(() => {
- {tab !== 'workflows' && }
+ {showToggle && }
diff --git a/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx b/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx
index 8d4c916c2a..e2da68ceb7 100644
--- a/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx
+++ b/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx
@@ -1,13 +1,11 @@
import { Box } from '@invoke-ai/ui-library';
import { ImageViewer } from 'features/gallery/components/ImageViewer/ImageViewer';
-import { useImageViewer } from 'features/gallery/components/ImageViewer/useImageViewer';
import { memo } from 'react';
const UpscalingTab = () => {
- const imageViewer = useImageViewer();
return (
- {imageViewer.isOpen && }
+
);
};