diff options
| author | Vespinian <vespinian@proton.me> | 2023-03-11 17:33:35 +0000 |
|---|---|---|
| committer | Vespinian <vespinian@proton.me> | 2023-03-11 17:33:35 +0000 |
| commit | 46f9fe3cd6b7be712d85cdfc2cdff7ac3fe878b5 (patch) | |
| tree | a134b22d856ab534c497b04c0d7ec87935cfcce7 /modules/codeformer_model.py | |
| parent | 2174f58daee1e077eec1125e196d34cc93dbaf23 (diff) | |
| parent | 94ffa9fc5386e51f20692ab46906135e8de33110 (diff) | |
| download | stable-diffusion-webui-gfx803-46f9fe3cd6b7be712d85cdfc2cdff7ac3fe878b5.tar.gz stable-diffusion-webui-gfx803-46f9fe3cd6b7be712d85cdfc2cdff7ac3fe878b5.tar.bz2 stable-diffusion-webui-gfx803-46f9fe3cd6b7be712d85cdfc2cdff7ac3fe878b5.zip | |
Merge branch 'master' of https://github.com/AUTOMATIC1111/stable-diffusion-webui
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 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:
|