ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-12-20 21:23:06 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Try to improve the sampling strategy a bit

Browse Source 
It sill fails sometimes when it does not sample a timestamp token for
the entire segment. We now print a message in such cases
This commit is contained in:
Georgi Gerganov 2022-10-17 23:52:24 +03:00
parent 20d8e7a309
commit d14823582d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
whisper.cpp
View File

@ -2425,7 +2425,7 @@ int whisper_full(
whisper_token id = 0;
whisper_token tid = whisper_token_beg(ctx);
id = whisper_sample_best(ctx, result_len == 0);
id = whisper_sample_best(ctx, result_len == 0 || i > 32);
if (i > 0) {
tid = whisper_sample_timestamp(ctx);
}
@ -2445,7 +2445,9 @@ int whisper_full(
// end of text token
if (id == whisper_token_eot(ctx)) {
if (result_len == 0) {
result_len = i + 1;
// TODO: figure out how to resolve this
fprintf(stderr, "\n%s: failed to generate timestamp token - this should not happen\n\n", __func__);
//result_len = i + 1;
}
break;
}