Update api.py (#979)

修复SynthesizerTrn kwargs 传递异常。
2025-04-05 04:22:46 +08:00 · 2024-04-17 17:33:01 +08:00 · 2024-04-17 17:33:01 +08:00 · 96b53be703
commit 96b53be703
parent 9c4ba08ccb
1 changed files with 2 additions and 1 deletions
--- a/api.py
+++ b/api.py
@ -174,11 +174,12 @@ def change_sovits_weights(sovits_path):
    hps = dict_s2["config"]
    hps = DictToAttrRecursive(hps)
    hps.model.semantic_frame_rate = "25hz"
+    model_params_dict = vars(hps.model)
    vq_model = SynthesizerTrn(
        hps.data.filter_length // 2 + 1,
        hps.train.segment_size // hps.data.hop_length,
        n_speakers=hps.data.n_speakers,
-        **hps.model
+        **model_params_dict
    )
    if ("pretrained" not in sovits_path):
        del vq_model.enc_q