lyuxiang.lx 1 年之前
父節點
當前提交
c939c80480
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      cosyvoice/llm/llm.py

+ 2 - 0
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)