extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-16 18:32:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Simplify onUpscaleClick (#602)

Browse Source 
* Simplified onUpscaleClick code.

* Updated fix with comment as to what it's fixing.

* Move the fix to enqueueImageVariationTask
This commit is contained in:
Marc-Andre Ferland 2022-12-05 00:16:10 -05:00 committed by GitHub
parent bd10a850fa
commit f701b8dc29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
ui/media/js/main.js
View File

@ -375,11 +375,21 @@ function onMakeSimilarClick(req, img) {
function enqueueImageVariationTask(req, img, reqDiff) {
const imageSeed = img.getAttribute('data-seed')
const newTaskRequest = modifyCurrentRequest(req, reqDiff, {
const newRequestBody = {
num_outputs: 1, // this can be user-configurable in the future
seed: imageSeed
})
}
// If the user is editing pictures, stop modifyCurrentRequest from importing
// new values by setting the missing properties to undefined
if (!('init_image' in req) && !('init_image' in reqDiff)) {
newRequestBody.init_image = undefined
newRequestBody.mask = undefined
} else if (!('mask' in req) && !('mask' in reqDiff)) {
newRequestBody.mask = undefined
}
const newTaskRequest = modifyCurrentRequest(req, reqDiff, newRequestBody)
newTaskRequest.numOutputsTotal = 1 // this can be user-configurable in the future
newTaskRequest.batchCount = 1
@ -387,28 +397,9 @@ function enqueueImageVariationTask(req, img, reqDiff) {
}
function onUpscaleClick(req, img) {
if (IMAGE_REGEX.test(req.init_image) && IMAGE_REGEX.test(req.mask)) {
enqueueImageVariationTask(req, img, {
use_upscale: upscaleModelField.value,
init_image: req.init_image,
mask: req.mask
})
}
else if (IMAGE_REGEX.test(req.init_image)) {
enqueueImageVariationTask(req, img, {
use_upscale: upscaleModelField.value,
init_image: req.init_image,
mask: undefined
})
}
else
{
enqueueImageVariationTask(req, img, {
use_upscale: upscaleModelField.value,
init_image: undefined,
mask: undefined
})
}
enqueueImageVariationTask(req, img, {
use_upscale: upscaleModelField.value
})
}
function onFixFacesClick(req, img) {