whisper : fix bug in token sampling logic

Could overflow buffer
3 years ago · 90513c3ac4
parent 0c710654a5
commit 90513c3ac4
1 changed files with 1 additions and 1 deletions
--- a/whisper.cpp
+++ b/whisper.cpp
@ -2470,7 +2470,7 @@ int whisper_full(
                        result_all.push_back({ t0, t1, 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--;