diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-05-29 06:34:26 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-06-13 09:38:28 +0000 |
| commit | 89352a2f52c6be51318192cedd86c8a342966a49 (patch) | |
| tree | fde621a62c1778d10aa1834597ec367de429642f /extensions-builtin/ScuNET | |
| parent | 59419bd64a1581caccaac04dceb66c1c069a2db1 (diff) | |
| download | stable-diffusion-webui-gfx803-89352a2f52c6be51318192cedd86c8a342966a49.tar.gz stable-diffusion-webui-gfx803-89352a2f52c6be51318192cedd86c8a342966a49.tar.bz2 stable-diffusion-webui-gfx803-89352a2f52c6be51318192cedd86c8a342966a49.zip | |
Move `load_file_from_url` to modelloader
Diffstat (limited to 'extensions-builtin/ScuNET')
| -rw-r--r-- | extensions-builtin/ScuNET/scripts/scunet_model.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/extensions-builtin/ScuNET/scripts/scunet_model.py b/extensions-builtin/ScuNET/scripts/scunet_model.py index 85b4505f..2785b551 100644 --- a/extensions-builtin/ScuNET/scripts/scunet_model.py +++ b/extensions-builtin/ScuNET/scripts/scunet_model.py @@ -6,12 +6,11 @@ import numpy as np import torch from tqdm import tqdm -from basicsr.utils.download_util import load_file_from_url - import modules.upscaler from modules import devices, modelloader, script_callbacks, errors from scunet_model_arch import SCUNet as net +from modules.modelloader import load_file_from_url from modules.shared import opts @@ -120,7 +119,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_download_path, file_name="%s.pth" % self.name, progress=True) + filename = load_file_from_url(self.model_url, model_dir=self.model_download_path, file_name=f"{self.name}.pth") else: filename = path if not os.path.exists(os.path.join(self.model_path, filename)) or filename is None: |