github/faster-whisper
1
1
Fork 0
mirror of https://github.com/SYSTRAN/faster-whisper.git synced 2026-01-09 21:48:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

Make batched suppress_tokens behaviour same as in sequential (#1194)

Browse Source
This commit is contained in:
Purfview
2024-12-11 11:51:38 +00:00
committed by GitHub
parent 8327d8cc64
commit f32c0e8af3
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
faster_whisper/transcribe.py
View File

@@ -495,7 +495,11 @@ class BatchedInferencePipeline:
initial_prompt=initial_prompt, initial_prompt=initial_prompt,
prefix=prefix, prefix=prefix,
suppress_blank=suppress_blank, suppress_blank=suppress_blank,
suppress_tokens=get_suppressed_tokens(tokenizer, suppress_tokens), suppress_tokens=(
get_suppressed_tokens(tokenizer, suppress_tokens)
if suppress_tokens
else suppress_tokens
),
prepend_punctuations=prepend_punctuations, prepend_punctuations=prepend_punctuations,
append_punctuations=append_punctuations, append_punctuations=append_punctuations,
max_new_tokens=max_new_tokens, max_new_tokens=max_new_tokens,