ggerganov commited on
Commit
5ccd239
·
1 Parent(s): 2094dd8

Update tests

Browse files
Files changed (2) hide show
  1. tests/CMakeLists.txt +6 -6
  2. 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;gh")
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;gh")
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
  }