feat: 添加导出 v3 的 script (#2208)

* feat: 添加导出 v3 的 script * Fix: 由于 export_torch_script_v3 的改动，v2 现在需要传入 top_k
2025-03-26 14:50:55 +08:00
parent f1332ff53a
commit b0e465eb72
4 changed files with 1275 additions and 9 deletions
--- a/GPT_SoVITS/f5_tts/model/backbones/dit.py
+++ b/GPT_SoVITS/f5_tts/model/backbones/dit.py
@@ -138,7 +138,7 @@ class DiT(nn.Module):
        time: float["b"] | float[""],  # time step  # noqa: F821 F722
            dt_base_bootstrap,
        text0,  # : int["b nt"]  # noqa: F722#####condition feature
-        use_grad_ckpt,  # bool
+        use_grad_ckpt=False,  # bool
        ###no-use
        drop_audio_cond=False,  # cfg for cond audio
        drop_text=False,  # cfg for text