Commit cefb5d6d authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix accessing options when they are not ready for SwinIR.

parent 8fba733c
Loading
Loading
Loading
Loading
+6 −2
Original line number Diff line number Diff line
@@ -99,11 +99,15 @@ class UpscalerSwinIR(Upscaler):
def upscale(
        img,
        model,
        tile=opts.SWIN_tile,
        tile_overlap=opts.SWIN_tile_overlap,
        tile=None,
        tile_overlap=None,
        window_size=8,
        scale=4,
):
    tile = tile or opts.SWIN_tile
    tile_overlap = tile_overlap or opts.SWIN_tile_overlap


    img = np.array(img)
    img = img[:, :, ::-1]
    img = np.moveaxis(img, 2, 0) / 255