Home Explore Help
Register Sign In
github
/
seamless_communication
mirror of https://github.com/facebookresearch/seamless_communication.git
2
0
Fork 0
Files Issues 0 Wiki
Browse Source

dimension increases for pretssel model output (#164)

Yilin Yang 1 year ago
parent
300d14efc5
commit
cd280d5245
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      src/seamless_communication/streaming/agents/pretssel_vocoder.py

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

@@ -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,