ソースを参照

more silent_token

lyuxiang.lx 1 ヶ月 前
コミット
cc91e40db8
1 ファイル変更1 行追加1 行削除
  1. 1 1
      cosyvoice/cli/model.py

+ 1 - 1
cosyvoice/cli/model.py

@@ -411,7 +411,7 @@ class CosyVoice3Model(CosyVoice2Model):
         self.llm_end_dict = {}
         self.hift_cache_dict = {}
         # FSQ silent token
-        self.silent_tokens = [2, 28, 29]
+        self.silent_tokens = [1, 2, 28, 29, 55, 2241, 2242, 2322, 2323]
 
     def token2wav(self, token, prompt_token, prompt_feat, embedding, token_offset, uuid, stream=False, finalize=False, speed=1.0):
         with torch.cuda.amp.autocast(self.fp16):