@@ -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