diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-05-29 09:41:36 +0300 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-06-13 12:38:29 +0300 |
| commit | 0afbc0c2355ead3a0ce7149a6d678f1f2e2fbfee (patch) | |
| tree | b6398b6d71ec3075926a4944d96f43415fc29d4e /extensions-builtin/SwinIR | |
| parent | 89352a2f52c6be51318192cedd86c8a342966a49 (diff) | |
Fix up `if "http" in ...:` to be more sensible startswiths
Diffstat (limited to 'extensions-builtin/SwinIR')
| -rw-r--r-- | extensions-builtin/SwinIR/scripts/swinir_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py index a5b0e2eb..4551761d 100644 --- a/extensions-builtin/SwinIR/scripts/swinir_model.py +++ b/extensions-builtin/SwinIR/scripts/swinir_model.py @@ -27,7 +27,7 @@ class UpscalerSwinIR(Upscaler): scalers = [] model_files = self.find_models(ext_filter=[".pt", ".pth"]) for model in model_files: - if "http" in model: + if model.startswith("http"): name = self.model_name else: name = modelloader.friendly_name(model) @@ -48,7 +48,7 @@ class UpscalerSwinIR(Upscaler): return img def load_model(self, path, scale=4): - if "http" in path: + if path.startswith("http"): filename = modelloader.load_file_from_url( url=path, model_dir=self.model_download_path, |