whisper : fix bug in token sampling logic

Could overflow buffer
pull/29/head
Georgi Gerganov 2 years ago
parent 877c058179
commit 332c9d77fe

@ -2470,7 +2470,7 @@ int whisper_full(
result_all.push_back({ t0, t1, text }); result_all.push_back({ t0, t1, text });
} }
text = ""; text = "";
while (result_cur[i].id > whisper_token_beg(ctx) && i < (int) result_cur.size()) { while (i < (int) result_cur.size() && result_cur[i].id > whisper_token_beg(ctx)) {
i++; i++;
} }
i--; i--;

Loading…
Cancel
Save