]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
whisper : perform entropy check only when we have at least 32 tokens (#412)
authorGeorgi Gerganov <redacted>
Wed, 18 Jan 2023 20:52:18 +0000 (22:52 +0200)
committerGeorgi Gerganov <redacted>
Wed, 18 Jan 2023 20:52:18 +0000 (22:52 +0200)
whisper.cpp

index 8846b8266ddb97bb40356e5695f865effad535d4..8543e4239d428fc301faebd9145cebe4fe96f961 100644 (file)
@@ -3721,7 +3721,7 @@ int whisper_full(
                     WHISPER_PRINT_DEBUG("%s: decoder %2d: score = %8.5f, result_len = %3d, avg_logprobs = %8.5f, entropy = %8.5f\n",
                             __func__, j, decoder.sequence.score, decoder.sequence.result_len, decoder.sequence.avg_logprobs, decoder.sequence.entropy);
 
-                    if (decoder.sequence.result_len > 8 && decoder.sequence.entropy < params.entropy_thold) {
+                    if (decoder.sequence.result_len > 32 && decoder.sequence.entropy < params.entropy_thold) {
                         WHISPER_PRINT_DEBUG("%s: decoder %2d: failed due to entropy %8.5f < %8.5f\n",
                                 __func__, j, decoder.sequence.entropy, params.entropy_thold);