File size: 506 Bytes
12a48eb
 
 
 
1e3934e
 
 
 
 
 
12a48eb
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from modules.live_portrait.model_downloader import download_model

MODELS_REALESRGAN_URL = {
    "realesr-general-x4v3": "https://huggingface.co/jhj0517/realesr-general-x4v3/resolve/main/realesr-general-x4v3.pth",
    "RealESRGAN_x2": "https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth",
}

MODELS_REALESRGAN_SCALABILITY = {
    "realesr-general-x4v3": [1, 2, 4],
    "RealESRGAN_x2": [2]
}


def download_resrgan_model(file_path, url):
    return download_model(file_path, url)