[debug] handle cache with prompt

2026-02-05 18:09:24 +08:00 · 2024-09-29 19:12:30 +08:00
parent c9acce1482
commit 8130abb5ea
2 changed files with 4 additions and 3 deletions
--- a/cosyvoice/flow/flow.py
+++ b/cosyvoice/flow/flow.py
@@ -141,6 +141,7 @@ class MaskedDiffWithXvec(torch.nn.Module):
            spks=embedding,
            cond=conds,
            n_timesteps=10,
+            prompt_len=mel_len1,
            required_cache_size=required_cache_size,
            flow_cache=flow_cache
        )