Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Decoding improvements #1033

Merged
merged 2 commits into from
Mar 6, 2023
Merged

Decoding improvements #1033

merged 2 commits into from
Mar 6, 2023

Conversation

jongwook
Copy link
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
1 participant