aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin/ScuNET
diff options
context:
space:
mode:
authorAarni Koskela <akx@iki.fi>2023-05-29 09:45:07 +0300
committerAarni Koskela <akx@iki.fi>2023-06-13 12:38:29 +0300
commite3a973a68df3cfe13039dae33d19cf2c02a741e0 (patch)
tree3ac4515ddedb6ec4ccc317dcc2aab2ea8bf49646 /extensions-builtin/ScuNET
parent0afbc0c2355ead3a0ce7149a6d678f1f2e2fbfee (diff)
Add TODO comments to sus model loads
Diffstat (limited to 'extensions-builtin/ScuNET')
-rw-r--r--extensions-builtin/ScuNET/scripts/scunet_model.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/extensions-builtin/ScuNET/scripts/scunet_model.py b/extensions-builtin/ScuNET/scripts/scunet_model.py
index 64f50829..da74a829 100644
--- a/extensions-builtin/ScuNET/scripts/scunet_model.py
+++ b/extensions-builtin/ScuNET/scripts/scunet_model.py
@@ -119,6 +119,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler):
def load_model(self, path: str):
device = devices.get_device_for('scunet')
if path.startswith("http"):
+ # TODO: this doesn't use `path` at all?
filename = load_file_from_url(self.model_url, model_dir=self.model_download_path, file_name=f"{self.name}.pth")
else:
filename = path