Add the possibility of using `return_logits_vocab` from Ctranslate2

SYSTRAN / faster-whisper

Faster Whisper transcription with CTranslate2

MIT License

12.59k stars 1.05k forks source link

Add the possibility of using `return_logits_vocab` from Ctranslate2 #1151

Open pierrepv8 opened 4 days ago

pierrepv8 commented 4 days ago

As latest version of Ctranslate2 generate method allow passing return_logits_vocab to include the log probs in its output. Would it be possible to expose a parameter to return_logits_vocab in FasterWhisperPipeline.transcribe() method ?