Merge pull request #8050 from Tpinion/master

Bugfix: Filter out temporary files that will be generated if the download fails.
3 years ago · 98695c1885
parent 7f2005127f ac4c7f05cd
commit 98695c1885
1 changed files with 1 additions and 1 deletions
--- a/modules/codeformer_model.py
+++ b/modules/codeformer_model.py
@ -55,7 +55,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, download_name='codeformer-v0.1.0.pth')
+                model_paths = modelloader.load_models(model_path, model_url, self.cmd_dir, download_name='codeformer-v0.1.0.pth', ext_filter=['.pth'])
                if len(model_paths) != 0:
                    ckpt_path = model_paths[0]
                else: