Spaces:
Running
Running
Update tests
Browse files- tests/CMakeLists.txt +6 -6
- whisper.cpp +1 -1
tests/CMakeLists.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
set(TEST_TARGET test-main-tiny)
|
| 2 |
add_test(NAME ${TEST_TARGET}
|
| 3 |
COMMAND $<TARGET_FILE:main>
|
| 4 |
-
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-tiny.bin
|
| 5 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 6 |
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "tiny;gh")
|
| 7 |
|
|
@@ -15,21 +15,21 @@ set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "tiny;en;gh")
|
|
| 15 |
set(TEST_TARGET test-main-base)
|
| 16 |
add_test(NAME ${TEST_TARGET}
|
| 17 |
COMMAND $<TARGET_FILE:main>
|
| 18 |
-
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-base.bin
|
| 19 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 20 |
-
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base
|
| 21 |
|
| 22 |
set(TEST_TARGET test-main-base.en)
|
| 23 |
add_test(NAME ${TEST_TARGET}
|
| 24 |
COMMAND $<TARGET_FILE:main>
|
| 25 |
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-base.en.bin
|
| 26 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 27 |
-
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base;en
|
| 28 |
|
| 29 |
set(TEST_TARGET test-main-small)
|
| 30 |
add_test(NAME ${TEST_TARGET}
|
| 31 |
COMMAND $<TARGET_FILE:main>
|
| 32 |
-
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-small.bin
|
| 33 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 34 |
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "small")
|
| 35 |
|
|
@@ -43,7 +43,7 @@ set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "small;en")
|
|
| 43 |
set(TEST_TARGET test-main-medium)
|
| 44 |
add_test(NAME ${TEST_TARGET}
|
| 45 |
COMMAND $<TARGET_FILE:main>
|
| 46 |
-
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-medium.bin
|
| 47 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 48 |
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "medium")
|
| 49 |
|
|
|
|
| 1 |
set(TEST_TARGET test-main-tiny)
|
| 2 |
add_test(NAME ${TEST_TARGET}
|
| 3 |
COMMAND $<TARGET_FILE:main>
|
| 4 |
+
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-tiny.bin -l fr
|
| 5 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 6 |
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "tiny;gh")
|
| 7 |
|
|
|
|
| 15 |
set(TEST_TARGET test-main-base)
|
| 16 |
add_test(NAME ${TEST_TARGET}
|
| 17 |
COMMAND $<TARGET_FILE:main>
|
| 18 |
+
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-base.bin -l fr
|
| 19 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 20 |
+
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base")
|
| 21 |
|
| 22 |
set(TEST_TARGET test-main-base.en)
|
| 23 |
add_test(NAME ${TEST_TARGET}
|
| 24 |
COMMAND $<TARGET_FILE:main>
|
| 25 |
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-base.en.bin
|
| 26 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 27 |
+
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base;en")
|
| 28 |
|
| 29 |
set(TEST_TARGET test-main-small)
|
| 30 |
add_test(NAME ${TEST_TARGET}
|
| 31 |
COMMAND $<TARGET_FILE:main>
|
| 32 |
+
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-small.bin -l fr
|
| 33 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 34 |
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "small")
|
| 35 |
|
|
|
|
| 43 |
set(TEST_TARGET test-main-medium)
|
| 44 |
add_test(NAME ${TEST_TARGET}
|
| 45 |
COMMAND $<TARGET_FILE:main>
|
| 46 |
+
-m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-medium.bin -l fr
|
| 47 |
-f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
|
| 48 |
set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "medium")
|
| 49 |
|
whisper.cpp
CHANGED
|
@@ -1012,7 +1012,7 @@ bool whisper_model_load(const std::string & fname, whisper_context & wctx) {
|
|
| 1012 |
|
| 1013 |
if (n_loaded == 0) {
|
| 1014 |
printf("%s: WARN no tensors loaded from model file - assuming empty model for testing\n", __func__);
|
| 1015 |
-
} else if (n_loaded != model.tensors.size()) {
|
| 1016 |
fprintf(stderr, "%s: ERROR not all tensors loaded from model file - expected %zu, got %d\n", __func__, model.tensors.size(), n_loaded);
|
| 1017 |
return false;
|
| 1018 |
}
|
|
|
|
| 1012 |
|
| 1013 |
if (n_loaded == 0) {
|
| 1014 |
printf("%s: WARN no tensors loaded from model file - assuming empty model for testing\n", __func__);
|
| 1015 |
+
} else if (n_loaded != (int) model.tensors.size()) {
|
| 1016 |
fprintf(stderr, "%s: ERROR not all tensors loaded from model file - expected %zu, got %d\n", __func__, model.tensors.size(), n_loaded);
|
| 1017 |
return false;
|
| 1018 |
}
|