diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-30 11:03:00 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-30 11:03:00 +0300 |
| commit | 2552204fcb04c13af93749bd125d086ba148517c (patch) | |
| tree | ce151cbf2a14621c3890bd027853f32fcc20da81 /modules/codeformer_model.py | |
| parent | aec811c5e6227e66a30e2d94261e7ea3037cd3fc (diff) | |
fix for broken codeformer in PR
Diffstat (limited to 'modules/codeformer_model.py')
| -rw-r--r-- | modules/codeformer_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
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:
|