Add files via upload

LiuHaiqiang123 · Dec 25, 2023 · 58a82fd · 58a82fd
1 parent 3becbbf
commit 58a82fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():