Update swinir.py

3 years ago · a5313351b3
parent 403bb02948
commit a5313351b3
1 changed files with 2 additions and 2 deletions
--- a/modules/swinir.py
+++ b/modules/swinir.py
@ -55,8 +55,8 @@ def load_models(dirname):
 def upscale(
    img,
    model,
-    tile=opts.GAN_tile,
-    tile_overlap=opts.GAN_tile_overlap,
+    tile=opts.SWIN_tile,
+    tile_overlap=opts.SWIN_tile_overlap,
    window_size=8,
    scale=4,
 ):