|
|
@ -71,6 +71,7 @@ struct whisper_params {
|
|
|
|
bool output_wts = false;
|
|
|
|
bool output_wts = false;
|
|
|
|
bool print_special = false;
|
|
|
|
bool print_special = false;
|
|
|
|
bool print_colors = false;
|
|
|
|
bool print_colors = false;
|
|
|
|
|
|
|
|
bool print_progress = false;
|
|
|
|
bool no_timestamps = false;
|
|
|
|
bool no_timestamps = false;
|
|
|
|
|
|
|
|
|
|
|
|
std::string language = "en";
|
|
|
|
std::string language = "en";
|
|
|
@ -112,6 +113,7 @@ bool whisper_params_parse(int argc, char ** argv, whisper_params & params) {
|
|
|
|
else if (arg == "-owts" || arg == "--output-words") { params.output_wts = true; }
|
|
|
|
else if (arg == "-owts" || arg == "--output-words") { params.output_wts = true; }
|
|
|
|
else if (arg == "-ps" || arg == "--print-special") { params.print_special = true; }
|
|
|
|
else if (arg == "-ps" || arg == "--print-special") { params.print_special = true; }
|
|
|
|
else if (arg == "-pc" || arg == "--print-colors") { params.print_colors = true; }
|
|
|
|
else if (arg == "-pc" || arg == "--print-colors") { params.print_colors = true; }
|
|
|
|
|
|
|
|
else if (arg == "-pp" || arg == "--print-progress") { params.print_progress = true; }
|
|
|
|
else if (arg == "-nt" || arg == "--no-timestamps") { params.no_timestamps = true; }
|
|
|
|
else if (arg == "-nt" || arg == "--no-timestamps") { params.no_timestamps = true; }
|
|
|
|
else if (arg == "-l" || arg == "--language") { params.language = argv[++i]; }
|
|
|
|
else if (arg == "-l" || arg == "--language") { params.language = argv[++i]; }
|
|
|
|
else if ( arg == "--prompt") { params.prompt = argv[++i]; }
|
|
|
|
else if ( arg == "--prompt") { params.prompt = argv[++i]; }
|
|
|
@ -150,6 +152,7 @@ void whisper_print_usage(int argc, char ** argv, const whisper_params & params)
|
|
|
|
fprintf(stderr, " -owts, --output-words [%-7s] output script for generating karaoke video\n", params.output_wts ? "true" : "false");
|
|
|
|
fprintf(stderr, " -owts, --output-words [%-7s] output script for generating karaoke video\n", params.output_wts ? "true" : "false");
|
|
|
|
fprintf(stderr, " -ps, --print-special [%-7s] print special tokens\n", params.print_special ? "true" : "false");
|
|
|
|
fprintf(stderr, " -ps, --print-special [%-7s] print special tokens\n", params.print_special ? "true" : "false");
|
|
|
|
fprintf(stderr, " -pc, --print-colors [%-7s] print colors\n", params.print_colors ? "true" : "false");
|
|
|
|
fprintf(stderr, " -pc, --print-colors [%-7s] print colors\n", params.print_colors ? "true" : "false");
|
|
|
|
|
|
|
|
fprintf(stderr, " -pp, --print-progress [%-7s] print progress\n", params.print_progress ? "true" : "false");
|
|
|
|
fprintf(stderr, " -nt, --no-timestamps [%-7s] do not print timestamps\n", params.no_timestamps ? "false" : "true");
|
|
|
|
fprintf(stderr, " -nt, --no-timestamps [%-7s] do not print timestamps\n", params.no_timestamps ? "false" : "true");
|
|
|
|
fprintf(stderr, " -l LANG, --language LANG [%-7s] spoken language\n", params.language.c_str());
|
|
|
|
fprintf(stderr, " -l LANG, --language LANG [%-7s] spoken language\n", params.language.c_str());
|
|
|
|
fprintf(stderr, " --prompt PROMPT [%-7s] initial prompt\n", params.prompt.c_str());
|
|
|
|
fprintf(stderr, " --prompt PROMPT [%-7s] initial prompt\n", params.prompt.c_str());
|
|
|
@ -601,7 +604,7 @@ int main(int argc, char ** argv) {
|
|
|
|
whisper_full_params wparams = whisper_full_default_params(WHISPER_SAMPLING_GREEDY);
|
|
|
|
whisper_full_params wparams = whisper_full_default_params(WHISPER_SAMPLING_GREEDY);
|
|
|
|
|
|
|
|
|
|
|
|
wparams.print_realtime = false;
|
|
|
|
wparams.print_realtime = false;
|
|
|
|
wparams.print_progress = false;
|
|
|
|
wparams.print_progress = params.print_progress;
|
|
|
|
wparams.print_timestamps = !params.no_timestamps;
|
|
|
|
wparams.print_timestamps = !params.no_timestamps;
|
|
|
|
wparams.print_special = params.print_special;
|
|
|
|
wparams.print_special = params.print_special;
|
|
|
|
wparams.translate = params.translate;
|
|
|
|
wparams.translate = params.translate;
|
|
|
|