Skip to content

Commit 0994506

Browse files
authored
cli : add no_speech_thold (#2663)
1 parent 53c9a3a commit 0994506

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

examples/cli/cli.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ struct whisper_params {
4343
float word_thold = 0.01f;
4444
float entropy_thold = 2.40f;
4545
float logprob_thold = -1.00f;
46+
float no_speech_thold = 0.6f;
4647
float grammar_penalty = 100.0f;
4748
float temperature = 0.0f;
4849
float temperature_inc = 0.2f;
@@ -135,6 +136,7 @@ static bool whisper_params_parse(int argc, char ** argv, whisper_params & params
135136
else if (arg == "-wt" || arg == "--word-thold") { params.word_thold = std::stof(argv[++i]); }
136137
else if (arg == "-et" || arg == "--entropy-thold") { params.entropy_thold = std::stof(argv[++i]); }
137138
else if (arg == "-lpt" || arg == "--logprob-thold") { params.logprob_thold = std::stof(argv[++i]); }
139+
else if (arg == "-nth" || arg == "--no-speech-thold") { params.no_speech_thold = std::stof(argv[++i]); }
138140
else if (arg == "-tp" || arg == "--temperature") { params.temperature = std::stof(argv[++i]); }
139141
else if (arg == "-tpi" || arg == "--temperature-inc") { params.temperature_inc = std::stof(argv[++i]); }
140142
else if (arg == "-debug"|| arg == "--debug-mode") { params.debug_mode = true; }
@@ -202,6 +204,7 @@ static void whisper_print_usage(int /*argc*/, char ** argv, const whisper_params
202204
fprintf(stderr, " -wt N, --word-thold N [%-7.2f] word timestamp probability threshold\n", params.word_thold);
203205
fprintf(stderr, " -et N, --entropy-thold N [%-7.2f] entropy threshold for decoder fail\n", params.entropy_thold);
204206
fprintf(stderr, " -lpt N, --logprob-thold N [%-7.2f] log probability threshold for decoder fail\n", params.logprob_thold);
207+
fprintf(stderr, " -nth N, --no-speech-thold N [%-7.2f] no speech threshold\n", params.no_speech_thold);
205208
fprintf(stderr, " -tp, --temperature N [%-7.2f] The sampling temperature, between 0 and 1\n", params.temperature);
206209
fprintf(stderr, " -tpi, --temperature-inc N [%-7.2f] The increment of temperature, between 0 and 1\n",params.temperature_inc);
207210
fprintf(stderr, " -debug, --debug-mode [%-7s] enable debug mode (eg. dump log_mel)\n", params.debug_mode ? "true" : "false");
@@ -1121,6 +1124,7 @@ int main(int argc, char ** argv) {
11211124

11221125
wparams.entropy_thold = params.entropy_thold;
11231126
wparams.logprob_thold = params.logprob_thold;
1127+
wparams.no_speech_thold = params.no_speech_thold;
11241128

11251129
wparams.no_timestamps = params.no_timestamps;
11261130

0 commit comments

Comments
 (0)