extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-08-13 17:57:20 +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
Download Patch File Download Diff File Expand all files Collapse all files

19
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,6 +414,8 @@ function getStartUpscaleHandler(reqBody, imageItemElem) {
} }
const imageElem = imageItemElem.querySelector('img') const imageElem = imageItemElem.querySelector('img')
const newTaskRequest = getCurrentUserRequest() const newTaskRequest = getCurrentUserRequest()
switch (mode) {
case 'img2img':
newTaskRequest.reqBody = Object.assign({}, reqBody, { newTaskRequest.reqBody = Object.assign({}, reqBody, {
num_outputs: 1, num_outputs: 1,
width: reqBody.width * 2, width: reqBody.width * 2,
@ -423,6 +425,17 @@ function getStartUpscaleHandler(reqBody, imageItemElem) {
prompt_strength: '0.5', prompt_strength: '0.5',
num_inference_steps: Math.min(100, reqBody.num_inference_steps * 2) 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