diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-19 09:09:00 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-19 09:09:18 +0300 |
| commit | df6fffb054f8d3444baa887151a4874506a68be1 (patch) | |
| tree | 6b2ab174215a1525a62d1fdd9fd46f7c61302e1d /extensions-builtin/ScuNET/scripts | |
| parent | 379fd6204dfc27e16acc03f705ecd9ff23c2d1c0 (diff) | |
change upscalers to download models into user-specified directory (from commandline args) rather than the default models/<...>
Diffstat (limited to 'extensions-builtin/ScuNET/scripts')
| -rw-r--r-- | extensions-builtin/ScuNET/scripts/scunet_model.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/extensions-builtin/ScuNET/scripts/scunet_model.py b/extensions-builtin/ScuNET/scripts/scunet_model.py index cc2cbc6a..45d9297b 100644 --- a/extensions-builtin/ScuNET/scripts/scunet_model.py +++ b/extensions-builtin/ScuNET/scripts/scunet_model.py @@ -121,8 +121,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler): def load_model(self, path: str): device = devices.get_device_for('scunet') if "http" in path: - filename = load_file_from_url(url=self.model_url, model_dir=self.model_path, file_name="%s.pth" % self.name, - progress=True) + filename = load_file_from_url(url=self.model_url, model_dir=self.model_download_path, file_name="%s.pth" % self.name, progress=True) else: filename = path if not os.path.exists(os.path.join(self.model_path, filename)) or filename is None: |