From 5251e75ab1d2fae2765a0d6c717cf9fcb7eec67b Mon Sep 17 00:00:00 2001 From: Ftps Date: Sun, 27 Aug 2023 19:12:30 +0900 Subject: [PATCH] fix path --- infer/lib/train/process_ckpt.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/infer/lib/train/process_ckpt.py b/infer/lib/train/process_ckpt.py index 37a8795..f2d73af 100644 --- a/infer/lib/train/process_ckpt.py +++ b/infer/lib/train/process_ckpt.py @@ -39,7 +39,7 @@ def savee(ckpt, sr, if_f0, name, epoch, version, hps): opt["sr"] = sr opt["f0"] = if_f0 opt["version"] = version - torch.save(opt, "weights/%s.pth" % name) + torch.save(opt, "assets/weights/%s.pth" % name) return "Success." except: return traceback.format_exc() @@ -182,7 +182,7 @@ def extract_small_model(path, name, sr, if_f0, info, version): opt["version"] = version opt["sr"] = sr opt["f0"] = int(if_f0) - torch.save(opt, "weights/%s.pth" % name) + torch.save(opt, "assets/weights/%s.pth" % name) return "Success." except: return traceback.format_exc() @@ -252,7 +252,7 @@ def merge(path1, path2, alpha1, sr, f0, info, name, version): opt["f0"] = 1 if f0 == i18n("是") else 0 opt["version"] = version opt["info"] = info - torch.save(opt, "weights/%s.pth" % name) + torch.save(opt, "assets/weights/%s.pth" % name) return "Success." except: return traceback.format_exc()