diff --git a/main.cpp b/main.cpp index d363ab7..9769b7f 100644 --- a/main.cpp +++ b/main.cpp @@ -28,6 +28,7 @@ std::string to_timestamp(int64_t t) { struct whisper_params { int32_t seed = -1; // RNG seed, not used currently int32_t n_threads = std::min(4, (int32_t) std::thread::hardware_concurrency()); + int32_t offset_ms = 0; bool verbose = false; bool translate = false; @@ -55,6 +56,8 @@ bool whisper_params_parse(int argc, char ** argv, whisper_params & params) { params.seed = std::stoi(argv[++i]); } else if (arg == "-t" || arg == "--threads") { params.n_threads = std::stoi(argv[++i]); + } else if (arg == "-o" || arg == "--offset") { + params.offset_ms = std::stoi(argv[++i]); } else if (arg == "-v" || arg == "--verbose") { params.verbose = true; } else if (arg == "--translate") { @@ -95,6 +98,7 @@ void whisper_print_usage(int argc, char ** argv, const whisper_params & params) fprintf(stderr, " -h, --help show this help message and exit\n"); fprintf(stderr, " -s SEED, --seed SEED RNG seed (default: -1)\n"); fprintf(stderr, " -t N, --threads N number of threads to use during computation (default: %d)\n", params.n_threads); + fprintf(stderr, " -o N, --offset N offset in milliseconds (default: %d)\n", params.offset_ms); fprintf(stderr, " -v, --verbose verbose output\n"); fprintf(stderr, " --translate translate from source language to english\n"); fprintf(stderr, " -ps, --print_special print special tokens\n"); @@ -203,6 +207,7 @@ int main(int argc, char ** argv) { wparams.translate = params.translate; wparams.language = params.language.c_str(); wparams.n_threads = params.n_threads; + wparams.offset_ms = params.offset_ms; if (whisper_full(ctx, wparams, pcmf32.data(), pcmf32.size()) != 0) { fprintf(stderr, "%s: failed to process audio\n", argv[0]); diff --git a/whisper.cpp b/whisper.cpp index 46a4caa..ca0c6a4 100644 --- a/whisper.cpp +++ b/whisper.cpp @@ -2256,6 +2256,7 @@ struct whisper_full_params whisper_full_default_params(enum whisper_decode_strat result = (struct whisper_full_params) { .strategy = WHISPER_DECODE_GREEDY, .n_threads = std::min(4, (int32_t) std::thread::hardware_concurrency()), + .offset_ms = 0, .translate = false, .print_special_tokens = false, @@ -2275,6 +2276,7 @@ struct whisper_full_params whisper_full_default_params(enum whisper_decode_strat result = (struct whisper_full_params) { .strategy = WHISPER_DECODE_GREEDY, .n_threads = std::min(4, (int32_t) std::thread::hardware_concurrency()), + .offset_ms = 0, .translate = false, .print_special_tokens = false, @@ -2329,7 +2331,7 @@ int whisper_full( int progress_step = 5; // main loop - int seek = 0; + int seek = params.offset_ms/10; while (true) { int progress_cur = (100*seek)/whisper_n_len(ctx); while (progress_cur >= progress_prev + progress_step) { diff --git a/whisper.h b/whisper.h index 2df5bdf..78e08b7 100644 --- a/whisper.h +++ b/whisper.h @@ -102,6 +102,7 @@ extern "C" { enum whisper_decode_strategy strategy; int n_threads; + int offset_ms; bool translate; bool print_special_tokens;