whisper : condition timestamps to be monotonically increasing

pull/425/head
Georgi Gerganov 2 years ago
parent 206fc93396
commit e20116c124
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

@ -2910,6 +2910,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)
{
const float logit_max = *std::max_element(logits.begin(), logits.end());

Loading…
Cancel
Save