extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-20 18:08:00 +02: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

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

@ -375,11 +375,21 @@ function onMakeSimilarClick(req, img) {
function enqueueImageVariationTask(req, img, reqDiff) { function enqueueImageVariationTask(req, img, reqDiff) {
const imageSeed = img.getAttribute('data-seed') const imageSeed = img.getAttribute('data-seed')
const newTaskRequest = modifyCurrentRequest(req, reqDiff, { const newRequestBody = {
num_outputs: 1, // this can be user-configurable in the future num_outputs: 1, // this can be user-configurable in the future
seed: imageSeed 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.numOutputsTotal = 1 // this can be user-configurable in the future
newTaskRequest.batchCount = 1 newTaskRequest.batchCount = 1
@ -387,29 +397,10 @@ function enqueueImageVariationTask(req, img, reqDiff) {
} }
function onUpscaleClick(req, img) { function onUpscaleClick(req, img) {
if (IMAGE_REGEX.test(req.init_image) && IMAGE_REGEX.test(req.mask)) {
enqueueImageVariationTask(req, img, { enqueueImageVariationTask(req, img, {
use_upscale: upscaleModelField.value, 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
})
}
}
function onFixFacesClick(req, img) { function onFixFacesClick(req, img) {
enqueueImageVariationTask(req, img, { enqueueImageVariationTask(req, img, {