PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-03-15 04:13:57 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

add merge message

Browse Source
This commit is contained in:
wataru 2023-05-11 23:20:08 +09:00
parent a598823573
commit 92caf3d5fc
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/voice_changer/RVC/modelMerger/MergeModel.py
View File

@ -49,10 +49,12 @@ def merge_model(request: MergeModelRequest):
merged: Dict[str, Any] = OrderedDict() merged: Dict[str, Any] = OrderedDict()
merged["weight"] = {} merged["weight"] = {}
print("merge start.")
for key in tqdm.tqdm(weights[0].keys()): for key in tqdm.tqdm(weights[0].keys()):
merged["weight"][key] = 0 merged["weight"][key] = 0
for i, weight in enumerate(weights): for i, weight in enumerate(weights):
merged["weight"][key] += weight[key] * alphas[i] merged["weight"][key] += weight[key] * alphas[i]
print("merge done. write metadata.")
merged["config"] = state_dict["config"] merged["config"] = state_dict["config"]
merged["params"] = state_dict["params"] if "params" in state_dict else None merged["params"] = state_dict["params"] if "params" in state_dict else None
@ -62,4 +64,5 @@ def merge_model(request: MergeModelRequest):
merged["embedder_name"] = ( merged["embedder_name"] = (
state_dict["embedder_name"] if "embedder_name" in state_dict else None state_dict["embedder_name"] if "embedder_name" in state_dict else None
) )
print("write metadata done.")
return merged return merged