back to single GFPGAN, fix some paths
This commit is contained in:
parent
03e06193eb
commit
c6ce43b00d
|
@ -24,9 +24,7 @@ sources: Dict[str, List[Tuple[str, str]]] = {
|
|||
'stabilityai/stable-diffusion-2-inpainting'),
|
||||
],
|
||||
'gfpgan': [
|
||||
('correction-gfpgan-v1-3-x2',
|
||||
'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth', 2),
|
||||
('correction-gfpgan-v1-3-x4',
|
||||
('correction-gfpgan-v1-3',
|
||||
'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth', 4),
|
||||
],
|
||||
'real_esrgan': [
|
||||
|
@ -57,7 +55,7 @@ def convert_real_esrgan(name: str, url: str, scale: int, opset: int):
|
|||
if not path.isfile(dest_path):
|
||||
print('PTH model not found, downloading...')
|
||||
download_path = load_file_from_url(
|
||||
url=url, model_dir=dest_path + '.cache', progress=True, file_name=None)
|
||||
url=url, model_dir=dest_path + '-cache', progress=True, file_name=None)
|
||||
copyfile(download_path, dest_path)
|
||||
|
||||
print('loading and training model')
|
||||
|
@ -106,7 +104,7 @@ def convert_gfpgan(name: str, url: str, scale: int, opset: int):
|
|||
if not path.isfile(dest_path):
|
||||
print('PTH model not found, downloading...')
|
||||
download_path = load_file_from_url(
|
||||
url=url, model_dir=dest_path + '.cache', progress=True, file_name=None)
|
||||
url=url, model_dir=dest_path + '-cache', progress=True, file_name=None)
|
||||
copyfile(download_path, dest_path)
|
||||
|
||||
print('loading and training model')
|
||||
|
|
|
@ -11,8 +11,7 @@ export const MODEL_LABELS = {
|
|||
'upscaling-real-esrgan-x4-plus': 'Real ESRGAN x4 Plus',
|
||||
'upscaling-real-esrgan-x4-v3': 'Real ESRGAN x4 v3',
|
||||
// correction
|
||||
'correction-gfpgan-v1-3-x2': 'GFPGAN v1.3 x2',
|
||||
'correction-gfpgan-v1-3-x4': 'GFPGAN v1.3 x4',
|
||||
'correction-gfpgan-v1-3': 'GFPGAN v1.3',
|
||||
|
||||
};
|
||||
|
||||
|
|
Loading…
Reference in New Issue