fix possible bug

2026-02-04 09:29:25 +08:00 · 2025-06-25 14:31:05 +08:00
parent 63856565f3
commit 46dfe0439b
1 changed files with 2 additions and 0 deletions
--- a/cosyvoice/flow/flow_matching.py
+++ b/cosyvoice/flow/flow_matching.py
@@ -127,6 +127,8 @@ class ConditionalCFM(BASECFM):
            return self.estimator(x, mask, mu, t, spks, cond, streaming=streaming)
        else:
            [estimator, stream], trt_engine = self.estimator.acquire_estimator()
+            # NOTE need to synchronize when switching stream
+            torch.cuda.current_stream().synchronize()
            with stream:
                estimator.set_input_shape('x', (2, 80, x.size(2)))
                estimator.set_input_shape('mask', (2, 1, x.size(2)))