diff options
| author | d8ahazard <d8ahazard@gmail.com> | 2022-09-29 19:59:53 -0500 |
|---|---|---|
| committer | d8ahazard <d8ahazard@gmail.com> | 2022-09-29 19:59:53 -0500 |
| commit | 435fd2112aee9a0e61408ac56663e41beea1e446 (patch) | |
| tree | 52ba152eb04e1c51f17cfbd6091b1b76b696fd7e /modules/swinir_model.py | |
| parent | d73741794d38a5c1aacacc7a6ed3fe3ca65724db (diff) | |
Fixes, cleanup.
Diffstat (limited to 'modules/swinir_model.py')
| -rw-r--r-- | modules/swinir_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/swinir_model.py b/modules/swinir_model.py index ea7b6301..41fda5a7 100644 --- a/modules/swinir_model.py +++ b/modules/swinir_model.py @@ -52,7 +52,7 @@ class UpscalerSwinIR(Upscaler): def load_model(self, path, scale=4): if "http" in path: - dl_name = "%s%s" % (self.name.replace(" ", "_"), ".pth") + dl_name = "%s%s" % (self.model_name.replace(" ", "_"), ".pth") filename = load_file_from_url(url=path, model_dir=self.model_path, file_name=dl_name, progress=True) else: filename = path |