From a2a0028bd464e6d937a5fef62feee53a0ef5c533 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 5 Feb 2023 15:53:43 -0600 Subject: [PATCH] feat(api): add CodeFormer to conversion script --- api/onnx_web/convert.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/api/onnx_web/convert.py b/api/onnx_web/convert.py index a01991aa..18c44865 100644 --- a/api/onnx_web/convert.py +++ b/api/onnx_web/convert.py @@ -56,6 +56,11 @@ base_models: Models = { "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": [ (