|
@@ -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:
|