fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-08 12:00:49 +08:00
Code Activity

feat: model embed author

Browse Source
This commit is contained in:
源文雨
2024-06-03 17:07:13 +09:00
parent ccecac6133
commit 07c66d3676
4 changed files with 309 additions and 291 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
infer/lib/train/process_ckpt.py
View File

@@ -43,6 +43,7 @@ def save_small_model(ckpt, sr, if_f0, name, epoch, version, hps):
opt["info"] = "%sepoch" % epoch
opt["name"] = name
opt["timestamp"] = int(time())
if hps.author: opt["author"] = hps.author
opt["sr"] = sr
opt["f0"] = if_f0
opt["version"] = version
@@ -55,7 +56,7 @@ def save_small_model(ckpt, sr, if_f0, name, epoch, version, hps):
return traceback.format_exc()
def extract_small_model(path, name, sr, if_f0, info, version):
def extract_small_model(path, name, author, sr, if_f0, info, version):
try:
ckpt = torch.load(path, map_location="cpu")
if "model" in ckpt:
@@ -178,6 +179,7 @@ def extract_small_model(path, name, sr, if_f0, info, version):
opt["info"] = info
opt["name"] = name
opt["timestamp"] = int(time())
if author: opt["author"] = author
opt["version"] = version
opt["sr"] = sr
opt["f0"] = int(if_f0)
@@ -214,6 +216,13 @@ def merge(path1, path2, alpha1, sr, f0, info, name, version):
continue
opt["weight"][key] = a[key]
return opt
def authors(c1, c2):
a1, a2 = c1.get("author", ""), c2.get("author", "")
if a1 == a2: return a1
if not a1: a1 = "Unknown"
if not a2: a2 = "Unknown"
return f"{a1} & {a2}"
ckpt1 = torch.load(path1, map_location="cpu")
ckpt2 = torch.load(path2, map_location="cpu")
@@ -242,8 +251,7 @@ def merge(path1, path2, alpha1, sr, f0, info, name, version):
opt["weight"][key] = (
alpha1 * (ckpt1[key].float()) + (1 - alpha1) * (ckpt2[key].float())
).half()
# except:
# pdb.set_trace()
author = authors(ckpt1, ckpt2)
opt["config"] = cfg
"""
if(sr=="40k"):opt["config"] = [1025, 32, 192, 192, 768, 2, 6, 3, 0, "1", [3, 7, 11], [[1, 3, 5], [1, 3, 5], [1, 3, 5]], [10, 10, 2, 2], 512, [16, 16, 4, 4,4], 109, 256, 40000]
@@ -252,6 +260,7 @@ def merge(path1, path2, alpha1, sr, f0, info, name, version):
"""
opt["name"] = name
opt["timestamp"] = int(time())
if author: opt["author"] = author
opt["sr"] = sr
opt["f0"] = 1 if f0 == i18n("是") else 0
opt["version"] = version