From 1aa28ddee1387d61bfd5b6ec4ada9b255112a0b0 Mon Sep 17 00:00:00 2001
From: Marc-Andre Ferland <madrang@gmail.com>
Date: Mon, 10 Oct 2022 22:29:15 -0400
Subject: [PATCH] Improved Redo and DoubleSize

---
 ui/media/main.js | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/ui/media/main.js b/ui/media/main.js
index 6764db77..7b6536a0 100644
--- a/ui/media/main.js
+++ b/ui/media/main.js
@@ -417,14 +417,23 @@ function getStartNewTaskHandler(reqBody, imageItemElem, mode) {
             case 'img2img_X2':
                 newTaskRequest.reqBody = Object.assign({}, reqBody, {
                     num_outputs: 1,
-                    init_image: imageElem.src,
-                    sampler: 'ddim',
                     prompt_strength: '0.5',
                 })
+                if (!newTaskRequest.reqBody.init_image || mode === 'img2img_X2') {
+                    newTaskRequest.reqBody.sampler = 'ddim'
+                    newTaskRequest.reqBody.init_image = imageElem.src
+                } else {
+                    newTaskRequest.reqBody.seed = 1 + newTaskRequest.reqBody.seed
+                }
                 if (mode === 'img2img_X2') {
                     newTaskRequest.reqBody.width = reqBody.width * 2
                     newTaskRequest.reqBody.height = reqBody.height * 2
                     newTaskRequest.reqBody.num_inference_steps = Math.min(100, reqBody.num_inference_steps * 2)
+                    if (useUpscalingField.checked) {
+                        newTaskRequest.reqBody.use_upscale = upscaleModelField.value
+                    } else {
+                        delete newTaskRequest.reqBody.use_upscale
+                    }
                 }
                 break
             case 'upscale':