jhj0517
commited on
Commit
·
692a471
1
Parent(s):
870f0f6
Update default model
Browse files
modules/image_restoration/model_downloader.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
from modules.live_portrait.model_downloader import download_model
|
2 |
|
3 |
MODELS_REALESRGAN_URL = {
|
|
|
4 |
"realesr-general-x4v3": "https://huggingface.co/jhj0517/realesr-general-x4v3/resolve/main/realesr-general-x4v3.pth",
|
5 |
}
|
6 |
|
|
|
1 |
from modules.live_portrait.model_downloader import download_model
|
2 |
|
3 |
MODELS_REALESRGAN_URL = {
|
4 |
+
"RealESRGAN_x2": "https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth",
|
5 |
"realesr-general-x4v3": "https://huggingface.co/jhj0517/realesr-general-x4v3/resolve/main/realesr-general-x4v3.pth",
|
6 |
}
|
7 |
|
modules/image_restoration/real_esrgan_inferencer.py
CHANGED
@@ -23,7 +23,7 @@ class RealESRGANInferencer:
|
|
23 |
|
24 |
def load_model(self,
|
25 |
model_name: Optional[str] = None,
|
26 |
-
scale: int =
|
27 |
progress: gr.Progress = gr.Progress()):
|
28 |
if model_name is None:
|
29 |
model_name = self.default_model
|
|
|
23 |
|
24 |
def load_model(self,
|
25 |
model_name: Optional[str] = None,
|
26 |
+
scale: int = 2,
|
27 |
progress: gr.Progress = gr.Progress()):
|
28 |
if model_name is None:
|
29 |
model_name = self.default_model
|