fix for broken codeformer in PR

This commit is contained in:
AUTOMATIC 2022-09-30 11:03:00 +03:00
parent aec811c5e6
commit 2552204fcb

View File

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