From 58a82fd20e5284460d20a6bf34224348bb5669a6 Mon Sep 17 00:00:00 2001 From: Hillobar <63615199+Hillobar@users.noreply.github.com> Date: Mon, 25 Dec 2023 11:01:53 -0800 Subject: [PATCH] Add files via upload --- rope/Coordinator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rope/Coordinator.py b/rope/Coordinator.py index 274f5e2c..1f81b288 100644 --- a/rope/Coordinator.py +++ b/rope/Coordinator.py @@ -208,7 +208,7 @@ def load_GPEN_256_model(): return session def load_GFPGAN_model(): - session = onnxruntime.InferenceSession( "./models/GPEN-BFR-512.onnx", providers=["CUDAExecutionProvider", 'CPUExecutionProvider']) + session = onnxruntime.InferenceSession( "./models/GFPGANv1.4.onnx", providers=["CUDAExecutionProvider", 'CPUExecutionProvider']) return session def load_codeformer_model():