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

Refactor the image button code

Browse Source
This commit is contained in:
cmdr2 2022-10-20 15:10:34 +05:30
parent 5d686b146d
commit 63e8614ace
1 changed files with 26 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -426,12 +426,20 @@ function onDownloadImageClick(req, img) {
imgDownload.click() imgDownload.click()
} }
function onMakeSimilarClick(req, img) { function modifyCurrentRequest(req, reqDiff) {
let newTaskRequest = getCurrentUserRequest() const newTaskRequest = getCurrentUserRequest()
newTaskRequest.reqBody = Object.assign({}, req, { newTaskRequest.reqBody = Object.assign({}, req, reqDiff, {
use_cpu: useCPUField.checked
})
newTaskRequest.seed = newTaskRequest.reqBody.seed
return newTaskRequest
}
function onMakeSimilarClick(req, img) {
const newTaskRequest = modifyCurrentRequest(req, {
num_outputs: 1, num_outputs: 1,
use_cpu: useCPUField.checked,
num_inference_steps: 50, num_inference_steps: 50,
guidance_scale: 7.5, guidance_scale: 7.5,
prompt_strength: 0.7, prompt_strength: 0.7,
@ -441,65 +449,42 @@ function onMakeSimilarClick(req, img) {
newTaskRequest.numOutputsTotal = 5 newTaskRequest.numOutputsTotal = 5
newTaskRequest.batchCount = 5 newTaskRequest.batchCount = 5
newTaskRequest.seed = newTaskRequest.reqBody.seed
delete newTaskRequest.reqBody.mask delete newTaskRequest.reqBody.mask
createTask(newTaskRequest) createTask(newTaskRequest)
} }
function onUpscaleClick(req, img) { function enqueueImageVariationTask(req, img, reqDiff) {
let newTaskRequest = getCurrentUserRequest()
const imageSeed = img.getAttribute('data-seed') const imageSeed = img.getAttribute('data-seed')
newTaskRequest.reqBody = Object.assign({}, req, { const newTaskRequest = modifyCurrentRequest(req, reqDiff, {
num_outputs: 1, num_outputs: 1, // this can be user-configurable in the future
use_cpu: useCPUField.checked,
use_upscale: upscaleModelField.value,
seed: imageSeed seed: imageSeed
}) })
newTaskRequest.numOutputsTotal = 1 newTaskRequest.numOutputsTotal = 1 // this can be user-configurable in the future
newTaskRequest.batchCount = 1 newTaskRequest.batchCount = 1
newTaskRequest.seed = newTaskRequest.reqBody.seed
createTask(newTaskRequest) createTask(newTaskRequest)
} }
function onUpscaleClick(req, img) {
enqueueImageVariationTask(req, img, {
use_upscale: upscaleModelField.value
})
}
function onFixFacesClick(req, img) { function onFixFacesClick(req, img) {
let newTaskRequest = getCurrentUserRequest() enqueueImageVariationTask(req, img, {
const imageSeed = img.getAttribute('data-seed') use_face_correction: 'GFPGANv1.3'
newTaskRequest.reqBody = Object.assign({}, req, {
num_outputs: 1,
use_cpu: useCPUField.checked,
use_face_correction: 'GFPGANv1.3',
seed: imageSeed
}) })
newTaskRequest.numOutputsTotal = 1
newTaskRequest.batchCount = 1
newTaskRequest.seed = newTaskRequest.reqBody.seed
createTask(newTaskRequest)
} }
function onContinueDrawingClick(req, img) { function onContinueDrawingClick(req, img) {
let newTaskRequest = getCurrentUserRequest() enqueueImageVariationTask(req, img, {
const imageSeed = img.getAttribute('data-seed') num_inference_steps: parseInt(req.num_inference_steps) + 25
newTaskRequest.reqBody = Object.assign({}, req, {
num_outputs: 1,
use_cpu: useCPUField.checked,
num_inference_steps: parseInt(req.num_inference_steps) + 25,
seed: imageSeed
}) })
newTaskRequest.numOutputsTotal = 1
newTaskRequest.batchCount = 1
newTaskRequest.seed = newTaskRequest.reqBody.seed
createTask(newTaskRequest)
} }
// makes a single image. don't call this directly, use makeImage() instead // makes a single image. don't call this directly, use makeImage() instead