Fix a bug that may cause size doesn't match error

2023-03-21 19:31:54 +08:00 · 2023-03-21 19:31:54 +08:00 · 306b70a749
parent b92442f60d
commit 306b70a749
1 changed files with 2 additions and 0 deletions
--- a/preprocess_flist_config.py
+++ b/preprocess_flist_config.py
@ -78,6 +78,8 @@ if __name__ == "__main__":
            f.write(wavpath + "\n")

    config_template["spk"] = spk_dict
+    config_template["model"]["n_speakers"] = spk_id
+	
    print("Writing configs/config.json")
    with open("configs/config.json", "w") as f:
        json.dump(config_template, f, indent=2)