feat: add additional Real ESRGAN models, strings for them
This commit is contained in:
parent
33eebacc0c
commit
d52c22e58b
|
@ -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),
|
||||
],
|
||||
}
|
||||
|
||||
|
|
|
@ -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> = {
|
||||
|
|
Loading…
Reference in New Issue