diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-11 14:58:45 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-11 14:58:45 +0300 |
| commit | 98695c18852512e6b1f6b0d7b0ab5e6e2f26f561 (patch) | |
| tree | 42ba041984d8c13fb568cfbe3607f7a4818cf0bd | |
| parent | 7f2005127ff20170e2d92f353416c7f0705c593b (diff) | |
| parent | ac4c7f05cd38dfa99cf64f7ddb9b1656e70a13c5 (diff) | |
Merge pull request #8050 from Tpinion/master
Bugfix: Filter out temporary files that will be generated if the download fails.
| -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 01fb7bd8..8d84bbc9 100644 --- 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:
|