Skip to content

Decoding improvements#1033

Merged
jongwook merged 2 commits intomainfrom
suppress_task_tokens
Mar 6, 2023
Merged

Decoding improvements#1033
jongwook merged 2 commits intomainfrom
suppress_task_tokens

Conversation

@jongwook
Copy link
Copy Markdown
Collaborator

@jongwook jongwook commented Mar 6, 2023

This fixes the failure cases where the model samples the task tokens (<|transcribe|> or <|translate|>) with a low probability but as reproducibly observed in #1009.

Additionally, the segmentation logic is improved to include the last segment in a DecodingResult even if it ends with a single timestamp.

In a future PR, we may also consider suppressing language tokens when it's not directly after a <|startoftext|> token

@jongwook jongwook changed the title suppress task tokens (transcribe/translate) Decoding improvements Mar 6, 2023
@jongwook jongwook merged commit eab8d92 into main Mar 6, 2023
@jongwook jongwook deleted the suppress_task_tokens branch March 6, 2023 22:01
zackees pushed a commit to zackees/whisper that referenced this pull request May 5, 2023
* suppress task tokens (transcribe/translate)

* not ignoring the last segment ending with one timestamp
ilanit1997 pushed a commit to ilanit1997/whisper that referenced this pull request May 16, 2023
* suppress task tokens (transcribe/translate)

* not ignoring the last segment ending with one timestamp
abyesilyurt pushed a commit to abyesilyurt/whisper that referenced this pull request Nov 13, 2023
* suppress task tokens (transcribe/translate)

* not ignoring the last segment ending with one timestamp
heejipark23 pushed a commit to heejipark23/whisper that referenced this pull request Sep 21, 2025
* suppress task tokens (transcribe/translate)

* not ignoring the last segment ending with one timestamp
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant