feat(api): add CodeFormer to conversion script

ssube · Feb 5, 2023 · a2a0028 · a2a0028
1 parent cb35907
commit a2a0028
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/api/onnx_web/convert.py b/api/onnx_web/convert.py
@@ -56,6 +56,11 @@
             "https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth",
             4,
         ),
+        (
+            "correction-codeformer",
+            "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth",
+            1,
+        )
     ],
     "upscaling": [
         (