extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-01-29 09:19:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

whisper : condition timestamps to be monotonically increasing (#425)

Browse Source
This commit is contained in:
Georgi Gerganov 2023-01-23 20:48:26 +02:00 committed by GitHub
parent ae16c21e9c
commit b5ddb16ec7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
whisper.cpp
View File

@ -2988,6 +2988,16 @@ static void whisper_process_logits(
} }
} }
// condition timestamp tokens to be increasing
// ref: https://github.com/openai/whisper/pull/831#issuecomment-1385910556
if (decoder.has_ts) {
const int tid0 = decoder.seek_delta/2;
for (int i = vocab.token_beg; i < vocab.token_beg + tid0; ++i) {
logits[i] = -INFINITY;
}
}
// populate the logprobs array (log_softmax) // populate the logprobs array (log_softmax)
{ {
const float logit_max = *std::max_element(logits.begin(), logits.end()); const float logit_max = *std::max_element(logits.begin(), logits.end());