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 = { const buttons = {
'imgUseBtn': { html: 'Use as Input', click: getUseAsInputHandler(imageItemElem) }, 'imgUseBtn': { html: 'Use as Input', click: getUseAsInputHandler(imageItemElem) },
'imgSaveBtn': { html: 'Download', click: getSaveImageHandler(imageItemElem, req['output_format']) }, '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) { 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 imgItemInfo = imageItemElem.querySelector('.imgItemInfo')
const createButton = function(name, btnInfo) { const createButton = function(name, btnInfo) {
@ -406,7 +406,7 @@ function getSaveImageHandler(imageItemElem, outputFormat) {
imgDownload.click() imgDownload.click()
} }
} }
function getStartUpscaleHandler(reqBody, imageItemElem) { function getStartNewTaskHandler(reqBody, imageItemElem, mode) {
return function() { return function() {
if (serverStatus !== 'online') { if (serverStatus !== 'online') {
alert('The server is still starting up..') alert('The server is still starting up..')
@ -414,15 +414,28 @@ function getStartUpscaleHandler(reqBody, imageItemElem) {
} }
const imageElem = imageItemElem.querySelector('img') const imageElem = imageItemElem.querySelector('img')
const newTaskRequest = getCurrentUserRequest() const newTaskRequest = getCurrentUserRequest()
newTaskRequest.reqBody = Object.assign({}, reqBody, { switch (mode) {
num_outputs: 1, case 'img2img':
width: reqBody.width * 2, newTaskRequest.reqBody = Object.assign({}, reqBody, {
height: reqBody.height * 2, num_outputs: 1,
init_image: imageElem.src, width: reqBody.width * 2,
sampler: 'ddim', height: reqBody.height * 2,
prompt_strength: '0.5', init_image: imageElem.src,
num_inference_steps: Math.min(100, reqBody.num_inference_steps * 2) 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.seed = newTaskRequest.reqBody.seed
newTaskRequest.numOutputsTotal = 1 newTaskRequest.numOutputsTotal = 1
newTaskRequest.batchCount = 1 newTaskRequest.batchCount = 1