Переглянути джерело

add missing prosody params (#141)

Pierre Andrews 1 рік тому
батько
коміт
6be7fd3468

+ 2 - 0
src/seamless_communication/inference/translator.py

@@ -366,6 +366,8 @@ class Translator(nn.Module):
                 text_generation_opts=text_generation_opts,
                 unit_generation_opts=unit_generation_opts,
                 bad_word_checker=self.bad_word_checker,
+                duration_factor=duration_factor,
+                prosody_encoder_input=prosody_encoder_input,
             )
 
         if output_modality == Modality.TEXT:

+ 4 - 0
src/seamless_communication/toxicity/mintox.py

@@ -150,6 +150,8 @@ def mintox_pipeline(
         beam_size=5, soft_max_seq_len=(25, 50)
     ),
     bad_word_checker: BadWordChecker = None,
+    duration_factor: float = 1.0,
+    prosody_encoder_input: Optional[SequenceData] = None,
 ) -> Tuple[SequenceToTextOutput, Optional[SequenceToUnitOutput]]:
     """MinTox: Mitigation at INference time of added TOXicity."""
     from seamless_communication.inference.translator import Modality, Translator
@@ -221,6 +223,8 @@ def mintox_pipeline(
             unit_generation_ngram_filtering=unit_generation_ngram_filtering,
             text_generation_opts=text_generation_opts,
             unit_generation_opts=unit_generation_opts,
+            duration_factor=duration_factor,
+            prosody_encoder_input=prosody_encoder_input,
         )
         batch_size = len(original_text_out.sentences)
         if batch_size > 1: