diff --git a/models/for-tests-silero-v5.1.2-ggml.bin b/models/for-tests-silero-v5.1.2-ggml.bin deleted file mode 100644 index c5ddfb53747..00000000000 Binary files a/models/for-tests-silero-v5.1.2-ggml.bin and /dev/null differ diff --git a/models/for-tests-silero-v6.2.0-ggml.bin b/models/for-tests-silero-v6.2.0-ggml.bin new file mode 100644 index 00000000000..e08fad67d94 Binary files /dev/null and b/models/for-tests-silero-v6.2.0-ggml.bin differ diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 0363193a745..09e77ea89c2 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -94,7 +94,7 @@ add_executable(${VAD_TEST} ${VAD_TEST}.cpp) target_include_directories(${VAD_TEST} PRIVATE ../include ../ggml/include ../examples) target_link_libraries(${VAD_TEST} PRIVATE common) target_compile_definitions(${VAD_TEST} PRIVATE - VAD_MODEL_PATH="${PROJECT_SOURCE_DIR}/models/for-tests-silero-v5.1.2-ggml.bin" + VAD_MODEL_PATH="${PROJECT_SOURCE_DIR}/models/for-tests-silero-v6.2.0-ggml.bin" SAMPLE_PATH="${PROJECT_SOURCE_DIR}/samples/jfk.wav") add_test(NAME ${VAD_TEST} COMMAND ${VAD_TEST}) set_tests_properties(${VAD_TEST} PROPERTIES LABELS "unit") @@ -106,7 +106,7 @@ target_include_directories(${VAD_TEST} PRIVATE ../include ../ggml/include ../exa target_link_libraries(${VAD_TEST} PRIVATE common) target_compile_definitions(${VAD_TEST} PRIVATE WHISPER_MODEL_PATH="${PROJECT_SOURCE_DIR}/models/ggml-base.en.bin" - VAD_MODEL_PATH="${PROJECT_SOURCE_DIR}/models/for-tests-silero-v5.1.2-ggml.bin" + VAD_MODEL_PATH="${PROJECT_SOURCE_DIR}/models/for-tests-silero-v6.2.0-ggml.bin" SAMPLE_PATH="${PROJECT_SOURCE_DIR}/samples/jfk.wav") add_test(NAME ${VAD_TEST} COMMAND ${VAD_TEST}) set_tests_properties(${VAD_TEST} PROPERTIES LABELS "base;en") diff --git a/tests/test-vad-full.cpp b/tests/test-vad-full.cpp index 3bba36b1668..8295b06d957 100644 --- a/tests/test-vad-full.cpp +++ b/tests/test-vad-full.cpp @@ -42,11 +42,13 @@ int main() { const int n_segments = whisper_full_n_segments(wctx); assert(n_segments == 1); + + printf("Segment text:\n%s", whisper_full_get_segment_text(wctx, 0)); assert(strcmp(" And so my fellow Americans, ask not what your country can do for you," " ask what you can do for your country.", whisper_full_get_segment_text(wctx, 0)) == 0); - assert(whisper_full_get_segment_t0(wctx, 0) == 29); - assert(whisper_full_get_segment_t1(wctx, 0) == 1049); + assert(whisper_full_get_segment_t0(wctx, 0) == 32); + assert(whisper_full_get_segment_t1(wctx, 0) == 1051); whisper_free(wctx); diff --git a/tests/test-vad.cpp b/tests/test-vad.cpp index 535721c8686..0003ae3acaa 100644 --- a/tests/test-vad.cpp +++ b/tests/test-vad.cpp @@ -36,7 +36,7 @@ struct whisper_vad_segments * test_detect_timestamps( struct whisper_vad_context * vctx, struct whisper_vad_params params) { struct whisper_vad_segments * timestamps = whisper_vad_segments_from_probs(vctx, params); - assert(whisper_vad_segments_n_segments(timestamps) == 5); + assert(whisper_vad_segments_n_segments(timestamps) == 4); for (int i = 0; i < whisper_vad_segments_n_segments(timestamps); ++i) { printf("VAD segment %d: start = %.2f, end = %.2f\n", i,