Pass patience and beam_size to faster-whisper.

2023-10-14 12:44:36 +02:00
parent 4cdce3b927
commit 14a7cab8eb
2 changed files with 3 additions and 1 deletions
@@ -145,6 +145,8 @@ class WhisperModel(faster_whisper.WhisperModel):
        result = self.model.generate(
                encoder_output,
                [prompt] * batch_size,
+                beam_size=options.beam_size,
+                patience=options.patience,
                length_penalty=options.length_penalty,
                max_length=self.max_length,
                suppress_blank=options.suppress_blank,