add llm export script

2026-02-05 09:59:23 +08:00 · 2024-08-28 18:21:11 +08:00
parent f4e70e222c
commit 9ab298dd49
9 changed files with 34 additions and 17 deletions
--- a/cosyvoice/transformer/decoder.py
+++ b/cosyvoice/transformer/decoder.py
@@ -174,7 +174,7 @@ class TransformerDecoder(torch.nn.Module):
                                                     memory_mask)
        return x

-    @torch.jit.ignore(drop=True)
+    @torch.jit.unused
    def forward_layers_checkpointed(self, x: torch.Tensor,
                                    tgt_mask: torch.Tensor,
                                    memory: torch.Tensor,