tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Merge branch 'dev' of github.com:litagin02/Style-Bert-VITS2

Browse Source
This commit is contained in:
tsukumi
2024-10-25 22:16:19 +09:00
parent d397a58874 a8b8c760ff
commit d762bf4401
6 changed files with 309 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
gradio_tabs/merge.py
View File

@@ -105,6 +105,8 @@ def merge_style_usual(
new_config = config_a.copy()
new_config["data"]["num_styles"] = len(new_style2id)
new_config["data"]["style2id"] = new_style2id
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
new_config["model_name"] = output_name
save_config(new_config, output_name)
@@ -160,6 +162,8 @@ def merge_style_add_diff(
new_config = config_a.copy()
new_config["data"]["num_styles"] = len(new_style2id)
new_config["data"]["style2id"] = new_style2id
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
new_config["model_name"] = output_name
save_config(new_config, output_name)
@@ -219,6 +223,8 @@ def merge_style_weighted_sum(
new_config = config_a.copy()
new_config["data"]["num_styles"] = len(new_style2id)
new_config["data"]["style2id"] = new_style2id
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
new_config["model_name"] = output_name
save_config(new_config, output_name)
@@ -268,6 +274,8 @@ def merge_style_add_null(
new_config = config_a.copy()
new_config["data"]["num_styles"] = len(new_style2id)
new_config["data"]["style2id"] = new_style2id
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
new_config["model_name"] = output_name
save_config(new_config, output_name)
@@ -362,6 +370,8 @@ def merge_models_usual(
new_config["model_name"] = output_name
new_config["data"]["num_styles"] = 1
new_config["data"]["style2id"] = {DEFAULT_STYLE: 0}
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
save_config(new_config, output_name)
neutral_vector_a = style_vectors_a[0]
@@ -444,6 +454,8 @@ def merge_models_add_diff(
new_config["model_name"] = output_name
new_config["data"]["num_styles"] = 1
new_config["data"]["style2id"] = {DEFAULT_STYLE: 0}
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
with open(assets_root / output_name / "config.json", "w", encoding="utf-8") as f:
json.dump(new_config, f, indent=2, ensure_ascii=False)
@@ -519,6 +531,8 @@ def merge_models_weighted_sum(
new_config["model_name"] = output_name
new_config["data"]["num_styles"] = 1
new_config["data"]["style2id"] = {DEFAULT_STYLE: 0}
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
with open(assets_root / output_name / "config.json", "w", encoding="utf-8") as f:
json.dump(new_config, f, indent=2, ensure_ascii=False)
@@ -595,6 +609,8 @@ def merge_models_add_null(
new_config["model_name"] = output_name
new_config["data"]["num_styles"] = 1
new_config["data"]["style2id"] = {DEFAULT_STYLE: 0}
if new_config["data"]["n_speakers"] == 1:
new_config["data"]["spk2id"] = { output_name : 0}
with open(assets_root / output_name / "config.json", "w", encoding="utf-8") as f:
json.dump(new_config, f, indent=2, ensure_ascii=False)