From c939c80480d64a7963b0a0dad253fddeb04294bb Mon Sep 17 00:00:00 2001
From: "lyuxiang.lx" <lyuxiang.lx@alibaba-inc.com>
Date: Sun, 1 Jun 2025 08:46:47 +0000
Subject: [PATCH] update

---
 cosyvoice/llm/llm.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cosyvoice/llm/llm.py b/cosyvoice/llm/llm.py
index a316e5d..d1fc777 100644
--- a/cosyvoice/llm/llm.py
+++ b/cosyvoice/llm/llm.py
@@ -441,6 +441,8 @@ class Qwen2LM(TransformerLM):
                     # in stream mode, yield token one by one
                     yield top_ids
                     out_tokens.append(top_ids)
+                    if len(out_tokens) == max_len:
+                        break
                 time.sleep(0.001)
             with self.lock:
                 self.vllm_output_queue.pop(uuid)