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 /modules/realesrgan_model.py | |
| 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 'modules/realesrgan_model.py')
| -rw-r--r-- | modules/realesrgan_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/realesrgan_model.py b/modules/realesrgan_model.py index 2d27b321..0d9c2e48 100644 --- a/modules/realesrgan_model.py +++ b/modules/realesrgan_model.py @@ -2,7 +2,6 @@ import os import numpy as np
from PIL import Image
-from basicsr.utils.download_util import load_file_from_url
from realesrgan import RealESRGANer
from modules.upscaler import Upscaler, UpscalerData
@@ -10,6 +9,7 @@ from modules.shared import cmd_opts, opts from modules import modelloader, errors
+
class UpscalerRealESRGAN(Upscaler):
def __init__(self, path):
self.name = "RealESRGAN"
@@ -71,7 +71,7 @@ class UpscalerRealESRGAN(Upscaler): return None
if info.local_data_path.startswith("http"):
- info.local_data_path = load_file_from_url(url=info.data_path, model_dir=self.model_download_path, progress=True)
+ info.local_data_path = modelloader.load_file_from_url(info.data_path, model_dir=self.model_download_path)
return info
except Exception:
|