diff --git a/infer-web.py b/infer-web.py index df2f9a4..beb8b9c 100644 --- a/infer-web.py +++ b/infer-web.py @@ -1668,7 +1668,7 @@ with gr.Blocks() as app: ckpt_b, alpha_a, sr_, - if_f0_, + 1 if if_f0_ == i18n("是") else 0, info__, name_to_save0, version_2, diff --git a/train/process_ckpt.py b/train/process_ckpt.py index 9d3d7e9..888051a 100644 --- a/train/process_ckpt.py +++ b/train/process_ckpt.py @@ -194,7 +194,7 @@ def change_info(path, info, name): return traceback.format_exc() -def merge(path1, path2, alpha1, sr, f0, info, name, version): +def merge(path1, path2, alpha1, sr, if_f0, info, name, version): try: def extract(ckpt): @@ -243,7 +243,7 @@ def merge(path1, path2, alpha1, sr, f0, info, name, version): elif(sr=="32k"):opt["config"] = [513, 32, 192, 192, 768, 2, 6, 3, 0, "1", [3, 7, 11], [[1, 3, 5], [1, 3, 5], [1, 3, 5]], [10, 4, 2, 2, 2], 512, [16, 16, 4, 4,4], 109, 256, 32000] """ opt["sr"] = sr - opt["f0"] = 1 if f0 == "是" else 0 + opt["f0"] = if_f0 opt["version"] = version opt["info"] = info torch.save(opt, "weights/%s.pth" % name)