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

Renamed getStartUpscaleHandler to getStartNewTaskHandler and added new tasks types.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-09 19:58:16 -04:00
parent 3e5f117066
commit a27c3f09b3
1 changed files with 25 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
ui/media/main.js
View File

@ -338,10 +338,10 @@ function showImages(req, 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: getStartUpscaleHandler(req, imageItemElem, 'img2img') },
'imgX2Btn': { html: 'ImgX2', click: getStartNewTaskHandler(req, imageItemElem, 'img2img') },
}
if (!req.use_upscale) {
buttons.upscaleBtn = { html: 'Upscale', click: getStartUpscaleHandler(req, imageItemElem, 'upscale') }
buttons.upscaleBtn = { html: 'Upscale', click: getStartNewTaskHandler(req, imageItemElem, 'upscale') }
}
const imgItemInfo = imageItemElem.querySelector('.imgItemInfo')
const createButton = function(name, btnInfo) {
@ -406,7 +406,7 @@ function getSaveImageHandler(imageItemElem, outputFormat) {
imgDownload.click()
}
}
function getStartUpscaleHandler(reqBody, imageItemElem) {
function getStartNewTaskHandler(reqBody, imageItemElem, mode) {
return function() {
if (serverStatus !== 'online') {
alert('The server is still starting up..')
@ -414,15 +414,28 @@ function getStartUpscaleHandler(reqBody, imageItemElem) {
}
const imageElem = imageItemElem.querySelector('img')
const newTaskRequest = getCurrentUserRequest()
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)
})
switch (mode) {
case 'img2img':
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)
})
break
case 'upscale':
newTaskRequest.reqBody = Object.assign({}, reqBody, {
num_outputs: 1,
//use_face_correction: 'GFPGANv1.3',
use_upscale: upscaleModelField.value,
})
break
default:
throw new Error("Unknown upscale mode: " + mode)
}
newTaskRequest.seed = newTaskRequest.reqBody.seed
newTaskRequest.numOutputsTotal = 1
newTaskRequest.batchCount = 1