]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
Update tests
authorGeorgi Gerganov <redacted>
Sat, 8 Oct 2022 08:34:20 +0000 (11:34 +0300)
committerGeorgi Gerganov <redacted>
Sat, 8 Oct 2022 08:43:42 +0000 (11:43 +0300)
tests/CMakeLists.txt
whisper.cpp

index bf4ea3515e71756e12ce923318af5319cfac9787..69b73ff88ed41f014525aafccab071cae7e1fd1e 100644 (file)
@@ -1,7 +1,7 @@
 set(TEST_TARGET test-main-tiny)
 add_test(NAME ${TEST_TARGET}
     COMMAND $<TARGET_FILE:main>
-    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-tiny.bin
+    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-tiny.bin -l fr
     -f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
 set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "tiny;gh")
 
@@ -15,21 +15,21 @@ set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "tiny;en;gh")
 set(TEST_TARGET test-main-base)
 add_test(NAME ${TEST_TARGET}
     COMMAND $<TARGET_FILE:main>
-    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-base.bin
+    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-base.bin -l fr
     -f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
-set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base;gh")
+set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base")
 
 set(TEST_TARGET test-main-base.en)
 add_test(NAME ${TEST_TARGET}
     COMMAND $<TARGET_FILE:main>
     -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-base.en.bin
     -f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
-set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base;en;gh")
+set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "base;en")
 
 set(TEST_TARGET test-main-small)
 add_test(NAME ${TEST_TARGET}
     COMMAND $<TARGET_FILE:main>
-    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-small.bin
+    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-small.bin -l fr
     -f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
 set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "small")
 
@@ -43,7 +43,7 @@ set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "small;en")
 set(TEST_TARGET test-main-medium)
 add_test(NAME ${TEST_TARGET}
     COMMAND $<TARGET_FILE:main>
-    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-medium.bin
+    -m ${PROJECT_SOURCE_DIR}/models/for-tests-ggml-medium.bin -l fr
     -f ${PROJECT_SOURCE_DIR}/samples/jfk.wav)
 set_tests_properties(${TEST_TARGET} PROPERTIES LABELS "medium")
 
index cdf76beb19f7016ef24df97386b1b8630993b4fa..af89815180910cb96dce8d351a9a7713694e4f3f 100644 (file)
@@ -1012,7 +1012,7 @@ bool whisper_model_load(const std::string & fname, whisper_context & wctx) {
 
         if (n_loaded == 0) {
             printf("%s: WARN no tensors loaded from model file - assuming empty model for testing\n", __func__);
-        } else if (n_loaded != model.tensors.size()) {
+        } else if (n_loaded != (int) model.tensors.size()) {
             fprintf(stderr, "%s: ERROR not all tensors loaded from model file - expected %zu, got %d\n", __func__, model.tensors.size(), n_loaded);
             return false;
         }