apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): add CodeFormer to conversion script

This commit is contained in:
Sean Sube 2023-02-05 15:53:43 -06:00
parent cb3590720b
commit a2a0028bd4
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/convert.py
View File

@ -56,6 +56,11 @@ base_models: Models = {
"https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth", "https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth",
4, 4,
), ),
(
"correction-codeformer",
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth",
1,
)
], ],
"upscaling": [ "upscaling": [
( (