public_git/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggml-org/whisper.cpp.git synced 2026-10-07 23:11:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

whisper : rename suppress_non_speech_tokens to suppress_nst (#2653)

Browse Source
This commit is contained in:
Georgi Gerganov
2024-12-21 12:54:35 +02:00
committed by GitHub
parent 944ce49439
commit f4668169a0
7 changed files with 31 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/whisper.cpp
+2 -2
View File
@@ -4676,7 +4676,7 @@ struct whisper_full_params whisper_full_default_params(enum whisper_sampling_str
/*.detect_language =*/ false,
/*.suppress_blank =*/ true,
/*.suppress_non_speech_tokens =*/ false,
/*.suppress_nst =*/ false,
/*.temperature =*/ 0.0f,
/*.max_initial_ts =*/ 1.0f,
@@ -4960,7 +4960,7 @@ static void whisper_process_logits(
// suppress non-speech tokens
// ref: https://github.com/openai/whisper/blob/7858aa9c08d98f75575035ecd6481f462d66ca27/whisper/tokenizer.py#L224-L253
if (params.suppress_non_speech_tokens) {
if (params.suppress_nst) {
for (const std::string & token : non_speech_tokens) {
const std::string suppress_tokens[] = {token, " " + token};
for (const std::string & suppress_token : suppress_tokens) {