From 11fa87388b77ccf4720de1ce1edf32fec3f103b6 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Wed, 30 Aug 2023 04:05:36 +1200
Subject: [PATCH] fix: Controlnet Prepreocessed Image Save Icon Missing

---
 .../components/ControlNetImagePreview.tsx      | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/invokeai/frontend/web/src/features/controlNet/components/ControlNetImagePreview.tsx b/invokeai/frontend/web/src/features/controlNet/components/ControlNetImagePreview.tsx
index 3641115c50..312d5a517b 100644
--- a/invokeai/frontend/web/src/features/controlNet/components/ControlNetImagePreview.tsx
+++ b/invokeai/frontend/web/src/features/controlNet/components/ControlNetImagePreview.tsx
@@ -180,11 +180,19 @@ const ControlNetImagePreview = ({ isSmall, controlNet }: Props) => {
           isUploadDisabled={true}
           isDropDisabled={!isEnabled}
         >
-          <IAIDndImageIcon
-            onClick={handleResetControlImage}
-            icon={controlImage ? <FaUndo /> : undefined}
-            tooltip="Reset Control Image"
-          />
+          <>
+            <IAIDndImageIcon
+              onClick={handleResetControlImage}
+              icon={controlImage ? <FaUndo /> : undefined}
+              tooltip="Reset Control Image"
+            />
+            <IAIDndImageIcon
+              onClick={handleSaveControlImage}
+              icon={controlImage ? <FaSave size={16} /> : undefined}
+              tooltip="Save Control Image"
+              styleOverrides={{ marginTop: 6 }}
+            />
+          </>
         </IAIDndImage>
       </Box>
       {pendingControlImages.includes(controlNetId) && (