extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-05-16 22:30:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

Splitted ImgX2 into Redo and Double Size.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-10 04:06:15 -04:00
parent 324c8f8146
commit 781effc34e
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ui/media/main.js
View File

@ -349,7 +349,8 @@ function showImages(reqBody, res, outputContainer, livePreview) {
const buttons = {
'imgUseBtn': { html: 'Use as Input', click: getUseAsInputHandler(imageItemElem) },
'imgSaveBtn': { html: 'Download', click: getSaveImageHandler(imageItemElem, req['output_format']) },
'imgX2Btn': { html: 'ImgX2', click: getStartNewTaskHandler(req, imageItemElem, 'img2img') },
'imgX2Btn': { html: 'Double Size', click: getStartNewTaskHandler(req, imageItemElem, 'img2img_X2') },
'imgRedoBtn': { html: 'Redo', click: getStartNewTaskHandler(req, imageItemElem, 'img2img') },
}
if (!req.use_upscale) {
buttons.upscaleBtn = { html: 'Upscale', click: getStartNewTaskHandler(req, imageItemElem, 'upscale') }
@ -413,15 +414,18 @@ function getStartNewTaskHandler(reqBody, imageItemElem, mode) {
const newTaskRequest = getCurrentUserRequest()
switch (mode) {
case 'img2img':
case 'img2img_X2':
newTaskRequest.reqBody = Object.assign({}, reqBody, {
num_outputs: 1,
width: reqBody.width * 2,
height: reqBody.height * 2,
init_image: imageElem.src,
sampler: 'ddim',
prompt_strength: '0.5',
num_inference_steps: Math.min(100, reqBody.num_inference_steps * 2)
})
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)
}
break
case 'upscale':
newTaskRequest.reqBody = Object.assign({}, reqBody, {