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
|