aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin
diff options
context:
space:
mode:
authorAarni Koskela <akx@iki.fi>2023-06-13 13:00:05 +0300
committerAarni Koskela <akx@iki.fi>2023-06-13 13:00:05 +0300
commit2667f47ffbf7c641a7e77abbdddf5e81bf144199 (patch)
treeeec30ca1808cd7a16ceb64d209c3b544a21defea /extensions-builtin
parentbf67a5dcf44c3dbd88d1913478d4e02477915f33 (diff)
Remove stray space from SwinIR model URL
Diffstat (limited to 'extensions-builtin')
-rw-r--r--extensions-builtin/SwinIR/scripts/swinir_model.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py
index 3ce622d9..c6bc53a8 100644
--- a/extensions-builtin/SwinIR/scripts/swinir_model.py
+++ b/extensions-builtin/SwinIR/scripts/swinir_model.py
@@ -11,6 +11,7 @@ from swinir_model_arch import SwinIR
from swinir_model_arch_v2 import Swin2SR
from modules.upscaler import Upscaler, UpscalerData
+SWINIR_MODEL_URL = "https://github.com/JingyunLiang/SwinIR/releases/download/v0.0/003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth"
device_swinir = devices.get_device_for('swinir')
@@ -18,9 +19,7 @@ device_swinir = devices.get_device_for('swinir')
class UpscalerSwinIR(Upscaler):
def __init__(self, dirname):
self.name = "SwinIR"
- self.model_url = "https://github.com/JingyunLiang/SwinIR/releases/download/v0.0" \
- "/003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR" \
- "-L_x4_GAN.pth "
+ self.model_url = SWINIR_MODEL_URL
self.model_name = "SwinIR 4x"
self.user_path = dirname
super().__init__()