whisper : fix segment length with params.no_timestamps == true

This commit is contained in:
Georgi Gerganov 2024-01-12 13:37:38 +02:00
parent f7908f9bb8
commit 6b01e3fedd
No known key found for this signature in database
GPG Key ID: BF970631944C16B7

View File

@ -5469,7 +5469,7 @@ int whisper_full_with_state(
(params.max_tokens > 0 && i >= params.max_tokens) || // max tokens per segment reached
(has_ts && seek + seek_delta + 100 >= seek_end) // end of audio reached
) {
if (result_len == 0) {
if (result_len == 0 && !params.no_timestamps) {
if (seek + seek_delta + 100 >= seek_end) {
result_len = i + 1;
} else {
@ -5479,7 +5479,7 @@ int whisper_full_with_state(
}
}
if (params.single_segment) {
if (params.single_segment || params.no_timestamps) {
result_len = i + 1;
seek_delta = 100*WHISPER_CHUNK_SIZE;
}