whisper : fix bug in token sampling logic

Could overflow buffer
ggerganov · Oct 7, 2022 · 90513c3 · 90513c3
1 parent 0c71065
commit 90513c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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--;