From 2552204fcb04c13af93749bd125d086ba148517c Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Fri, 30 Sep 2022 11:03:00 +0300
Subject: fix for broken codeformer in PR

---
 modules/codeformer_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'modules/codeformer_model.py')

diff --git a/modules/codeformer_model.py b/modules/codeformer_model.py
index efd881eb..a29f3855 100644
--- a/modules/codeformer_model.py
+++ b/modules/codeformer_model.py
@@ -54,7 +54,7 @@ def setup_model(dirname):
                 if self.net is not None and self.face_helper is not None:
                     self.net.to(devices.device_codeformer)
                     return self.net, self.face_helper
-                model_paths = modelloader.load_models(model_path, model_url, self.cmd_dir)
+                model_paths = modelloader.load_models(model_path, model_url, self.cmd_dir, download_name='codeformer-v0.1.0.pth')
                 if len(model_paths) != 0:
                     ckpt_path = model_paths[0]
                 else:
-- 
cgit v1.2.1