diff --git a/ui/media/main.js b/ui/media/main.js index a3f29ed2..da28d7bb 100644 --- a/ui/media/main.js +++ b/ui/media/main.js @@ -449,7 +449,10 @@ function getStartNewTaskHandler(reqBody, imageItemElem, mode) { switch (mode) { case 'img2img': case 'img2img_X2': - newTaskRequest.reqBody = Object.assign({}, reqBody, { num_outputs: 1 }) + newTaskRequest.reqBody = Object.assign({}, reqBody, { + num_outputs: 1, + use_cpu: useCPUField.checked, + }) if (!newTaskRequest.reqBody.init_image || mode === 'img2img_X2') { newTaskRequest.reqBody.sampler = 'ddim' newTaskRequest.reqBody.prompt_strength = '0.5'