瀏覽代碼

feat:修改generate_long的并发逻辑

zhaohaipeng 2 月之前
父節點
當前提交
ecb04ca8d1
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      fish_speech/models/text2semantic/inference.py

+ 1 - 1
fish_speech/models/text2semantic/inference.py

@@ -995,7 +995,7 @@ def generate_long_batched(
 
         results = {}
 
-        with ThreadPoolExecutor(max_workers=3) as executor:
+        with ThreadPoolExecutor(max_workers=min(3, len(batches))) as executor:
             futures = [
                 executor.submit(run_one_batch, i, b)
                 for i, b in enumerate(batches)