apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

back to single GFPGAN, fix some paths

This commit is contained in:
Sean Sube 2023-01-16 21:55:54 -06:00
parent 03e06193eb
commit c6ce43b00d
2 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/convert.py
View File

@ -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')

3
gui/src/strings.ts
View File

@ -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',
};