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

SYSTRAN · Dec 11, 2024 · f32c0e8 · f32c0e8
1 parent 8327d8c
commit f32c0e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py
@@ -495,7 +495,11 @@ def transcribe(
             initial_prompt=initial_prompt,
             prefix=prefix,
             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,
             append_punctuations=append_punctuations,
             max_new_tokens=max_new_tokens,