Fix "detect image size" button

This commit is contained in:
missionfloyd
2023-12-09 20:02:03 -07:00
parent 2daf98a5b6
commit a8e41f585e
2 changed files with 5 additions and 5 deletions

View File

@@ -317,9 +317,9 @@ function selectCheckpoint(name) {
gradioApp().getElementById('change_checkpoint').click();
}
function currentImg2imgSourceResolution(w, h, scaleBy) {
var img = gradioApp().querySelector('#mode_img2img > div[style="display: block;"] img');
return img ? [img.naturalWidth, img.naturalHeight, scaleBy] : [0, 0, scaleBy];
function currentImg2imgSourceResolution(w, h) {
var img = gradioApp().querySelector('#mode_img2img > div[style="display: block;"] :is(img, canvas)');
return img ? [img.naturalWidth || img.width, img.naturalHeight || img.height] : [0, 0];
}
function updateImg2imgResizeToTextAfterChangingImage() {