ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-04-20 17:11:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

whisper : add check for max number of decoders

Browse Source
This commit is contained in:
Georgi Gerganov 2023-11-15 13:28:16 +02:00
parent 6c8a003acf
commit 820f45895e
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
whisper.cpp
View File

@ -148,7 +148,7 @@ static void whisper_log_callback_default(ggml_log_level level, const char * text
//#define WHISPER_USE_FLASH_ATTN
//#define WHISPER_USE_FLASH_FF
#define WHISPER_MAX_DECODERS 16
#define WHISPER_MAX_DECODERS 8
#define WHISPER_MAX_NODES 4096
//
@ -5075,6 +5075,11 @@ int whisper_full_with_state(
n_decoders = std::max(1, n_decoders);
if (n_decoders > WHISPER_MAX_DECODERS) {
WHISPER_LOG_ERROR("%s: too many decoders requested (%d), max = %d\n", __func__, n_decoders, WHISPER_MAX_DECODERS);
return -4;
}
// TAGS: WHISPER_DECODER_INIT
for (int j = 1; j < n_decoders; j++) {
auto & decoder = state->decoders[j];