Commit a5e7b371 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix the bug with broken rescaling in PR

parent 8f1b3153
Loading
Loading
Loading
Loading
+11 −3
Original line number Diff line number Diff line
@@ -219,9 +219,17 @@ def resize_image(resize_mode, im, width, height):
        if opts.upscaler_for_img2img is None or opts.upscaler_for_img2img == "None" or im.mode == 'L':
            return im.resize((w, h), resample=LANCZOS)

        upscaler = [x for x in shared.sd_upscalers if x.name == opts.upscaler_for_img2img][0]
        scale = w / im.width
        return upscaler.scaler.upscale(im, scale)
        upscalers = [x for x in shared.sd_upscalers if x.name == opts.upscaler_for_img2img]
        assert len(upscalers) > 0, f"could not find upscaler named {opts.upscaler_for_img2img}"

        upscaler = upscalers[0]
        scale = max(w / im.width, h / im.height)
        upscaled = upscaler.scaler.upscale(im, scale)

        if upscaled.width != w or upscaled.height != h:
            upscaled = im.resize((w, h), resample=LANCZOS)

        return upscaled

    if resize_mode == 0:
        res = resize(im, width, height)
+1 −1
Original line number Diff line number Diff line
@@ -60,7 +60,7 @@ class Upscaler:
                break
            img = self.do_upscale(img, selected_model)
        if img.width != dest_w or img.height != dest_h:
            img = img.resize((dest_w, dest_h), resample=LANCZOS)
            img = img.resize((int(dest_w), int(dest_h)), resample=LANCZOS)

        return img