extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-01-28 00:39:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix

Browse Source 
Longer prompts could cause out-of-bounds access
This commit is contained in:
Georgi Gerganov 2022-09-30 20:37:29 +03:00
parent 3bcdbdfc32
commit 15b49e8baf
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.cpp
View File

@ -1859,7 +1859,7 @@ whisper_vocab::id whisper_sample_best(
if (need_timestamp) { if (need_timestamp) {
// at the end of the 30-second audio segment, we start giving preference to time tokens // at the end of the 30-second audio segment, we start giving preference to time tokens
for (int i = 0; i < top_k; i++) { for (int i = 0; i < top_k; i++) {
if (probs_id[i].second > vocab.token_beg + 1300 && probs_id[i].first > probs_id[0].first*0.1) { if (probs_id[i].second > vocab.token_beg + 1300 && probs_id[i].first > 0.01*probs_id[0].first) {
return probs_id[i].second; return probs_id[i].second;
} }
} }
@ -2201,7 +2201,7 @@ int main(int argc, char ** argv) {
} }
// the generated text including timestamps // the generated text including timestamps
std::vector<whisper_result> result_all; //std::vector<whisper_result> result_all;
// main loop // main loop
int seek = 0; int seek = 0;
@ -2258,7 +2258,7 @@ int main(int argc, char ** argv) {
int result_len = 0; int result_len = 0;
std::vector<whisper_result> result_cur; std::vector<whisper_result> result_cur;
for (int i = 0; i < model.hparams.n_text_ctx/2; ++i) { for (int i = 0; i < model.hparams.n_text_ctx/2 - 4; ++i) {
// decode // decode
if (prompt.size() > 0) { if (prompt.size() > 0) {
const int64_t t_start_us = ggml_time_us(); const int64_t t_start_us = ggml_time_us();
@ -2323,7 +2323,7 @@ int main(int argc, char ** argv) {
} }
result_cur.resize(result_len); result_cur.resize(result_len);
result_all.insert(result_all.end(), result_cur.begin(), result_cur.end()); //result_all.insert(result_all.end(), result_cur.begin(), result_cur.end());
for (const auto & r : result_cur) { for (const auto & r : result_cur) {
prompt_past.push_back(r.id); prompt_past.push_back(r.id);