apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat: add additional Real ESRGAN models, strings for them

This commit is contained in:
Sean Sube 2023-01-16 20:38:16 -06:00
parent 33eebacc0c
commit d52c22e58b
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/convert.py
View File

@ -28,8 +28,12 @@ sources: Dict[str, List[Tuple[str, str]]] = {
'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth'),
],
'real_esrgan': [
('upscaling-real-esrgan-x2-plus',
'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth', 2)
('upscaling-real-esrgan-x4-plus',
'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth', 4),
('upscaling-real-esrgan-x4-v3',
'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth', 4),
],
}

8
gui/src/strings.ts
View File

@ -2,8 +2,16 @@
export const MODEL_LABELS = {
'stable-diffusion-onnx-v1-4': 'Stable Diffusion v1.4',
'stable-diffusion-onnx-v1-5': 'Stable Diffusion v1.5',
'stable-diffusion-onnx-v1-inpainting': 'SD Inpainting v1',
'stable-diffusion-onnx-v2-0': 'Stable Diffusion v2.0',
'stable-diffusion-onnx-v2-1': 'Stable Diffusion v2.1',
'stable-diffusion-onnx-v2-inpainting': 'SD Inpainting v2',
// upscaling
'upscaling-real-esrgan-x2-plus': 'Real ESRGAN x2 Plus',
'upscaling-real-esrgan-x4-plus': 'Real ESRGAN x4 Plus',
'upscaling-real-esrgan-x4-v3': 'Real ESRGAN x4 v3',
// correction
'correction-gfpgan-v1-3': 'GFPGAN v1.3',
};
export const PLATFORM_LABELS: Record<string, string> = {