whisper : rename suppress_non_speech_tokens to suppress_nst (#2653)

2025-06-18 14:58:09 +00:00 · 2024-12-21 12:54:35 +02:00
parent 944ce49439
commit f4668169a0
7 changed files with 31 additions and 27 deletions
--- a/src/whisper.cpp
+++ b/src/whisper.cpp
@ -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) {