Loading modules/codeformer_model.py +1 −1 Original line number Original line Diff line number Diff line Loading @@ -55,7 +55,7 @@ def setup_model(dirname): if self.net is not None and self.face_helper is not None: if self.net is not None and self.face_helper is not None: self.net.to(devices.device_codeformer) self.net.to(devices.device_codeformer) return self.net, self.face_helper 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: if len(model_paths) != 0: ckpt_path = model_paths[0] ckpt_path = model_paths[0] else: else: Loading Loading
modules/codeformer_model.py +1 −1 Original line number Original line Diff line number Diff line Loading @@ -55,7 +55,7 @@ def setup_model(dirname): if self.net is not None and self.face_helper is not None: if self.net is not None and self.face_helper is not None: self.net.to(devices.device_codeformer) self.net.to(devices.device_codeformer) return self.net, self.face_helper 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: if len(model_paths) != 0: ckpt_path = model_paths[0] ckpt_path = model_paths[0] else: else: Loading
