diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-19 06:09:00 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-19 06:09:18 +0000 |
| commit | df6fffb054f8d3444baa887151a4874506a68be1 (patch) | |
| tree | 6b2ab174215a1525a62d1fdd9fd46f7c61302e1d /extensions-builtin/SwinIR/scripts/swinir_model.py | |
| parent | 379fd6204dfc27e16acc03f705ecd9ff23c2d1c0 (diff) | |
| download | stable-diffusion-webui-gfx803-df6fffb054f8d3444baa887151a4874506a68be1.tar.gz stable-diffusion-webui-gfx803-df6fffb054f8d3444baa887151a4874506a68be1.tar.bz2 stable-diffusion-webui-gfx803-df6fffb054f8d3444baa887151a4874506a68be1.zip | |
change upscalers to download models into user-specified directory (from commandline args) rather than the default models/<...>
Diffstat (limited to 'extensions-builtin/SwinIR/scripts/swinir_model.py')
| -rw-r--r-- | extensions-builtin/SwinIR/scripts/swinir_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py index 0ba50487..1c7bf325 100644 --- a/extensions-builtin/SwinIR/scripts/swinir_model.py +++ b/extensions-builtin/SwinIR/scripts/swinir_model.py @@ -51,7 +51,7 @@ class UpscalerSwinIR(Upscaler): def load_model(self, path, scale=4): if "http" in path: 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) + filename = load_file_from_url(url=path, model_dir=self.model_download_path, file_name=dl_name, progress=True) else: filename = path if filename is None or not os.path.exists(filename): |