Fixes, cleanup.

This commit is contained in:
d8ahazard
2022-09-29 19:59:53 -05:00
parent d73741794d
commit 435fd2112a
3 changed files with 5 additions and 17 deletions

View File

@@ -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