浏览代码

dimension increases for pretssel model output (#164)

Yilin Yang 1 年之前
父节点
当前提交
cd280d5245
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/seamless_communication/streaming/agents/pretssel_vocoder.py

+ 1 - 1
src/seamless_communication/streaming/agents/pretssel_vocoder.py

@@ -95,7 +95,7 @@ class PretsselVocoderAgent(NoUpdateTargetMixin, TextToSpeechAgent):
 
         return WriteAction(
             SpeechSegment(
-                content=wav[0][0].tolist(),
+                content=wav[0][0][0].tolist(),
                 finished=states.source_finished,
                 sample_rate=self.vocoder_sample_rate,
                 tgt_lang=tgt_lang,