소스 검색

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,