diff options
author | Aarni Koskela <akx@iki.fi> | 2023-05-29 06:45:07 +0000 |
---|---|---|
committer | Aarni Koskela <akx@iki.fi> | 2023-06-13 09:38:29 +0000 |
commit | e3a973a68df3cfe13039dae33d19cf2c02a741e0 (patch) | |
tree | 3ac4515ddedb6ec4ccc317dcc2aab2ea8bf49646 | |
parent | 0afbc0c2355ead3a0ce7149a6d678f1f2e2fbfee (diff) | |
download | stable-diffusion-webui-gfx803-e3a973a68df3cfe13039dae33d19cf2c02a741e0.tar.gz stable-diffusion-webui-gfx803-e3a973a68df3cfe13039dae33d19cf2c02a741e0.tar.bz2 stable-diffusion-webui-gfx803-e3a973a68df3cfe13039dae33d19cf2c02a741e0.zip |
Add TODO comments to sus model loads
-rw-r--r-- | extensions-builtin/ScuNET/scripts/scunet_model.py | 1 | ||||
-rw-r--r-- | modules/esrgan_model.py | 1 |
2 files changed, 2 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 diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py index 0666a2c2..a20e8d91 100644 --- a/modules/esrgan_model.py +++ b/modules/esrgan_model.py @@ -151,6 +151,7 @@ class UpscalerESRGAN(Upscaler): def load_model(self, path: str):
if path.startswith("http"):
+ # TODO: this doesn't use `path` at all?
filename = modelloader.load_file_from_url(
url=self.model_url,
model_dir=self.model_download_path,
|