]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
java : fix test (#1492)
authorGeorgi Gerganov <redacted>
Wed, 15 Nov 2023 15:42:53 +0000 (17:42 +0200)
committerGitHub <redacted>
Wed, 15 Nov 2023 15:42:53 +0000 (17:42 +0200)
bindings/java/src/test/java/io/github/ggerganov/whispercpp/WhisperCppTest.java

index 2973243de497f34074f59802db90df91db6b554b..034726ad29c972d907f3fe6f7a5c9ece42cf1a6f 100644 (file)
@@ -45,7 +45,7 @@ class WhisperCppTest {
         assertEquals(16384, params.n_max_text_ctx);\r
         assertFalse(params.translate);\r
         assertEquals(0.01f, params.thold_pt);\r
-        assertEquals(2, params.beam_search.beam_size);\r
+        assertEquals(5, params.beam_search.beam_size);\r
         assertEquals(-1.0f, params.beam_search.patience);\r
     }\r
 \r
@@ -58,7 +58,7 @@ class WhisperCppTest {
         assertEquals(WhisperSamplingStrategy.WHISPER_SAMPLING_GREEDY.ordinal(), params.strategy);\r
         assertNotEquals(0, params.n_threads);\r
         assertEquals(16384, params.n_max_text_ctx);\r
-        assertEquals(2, params.greedy.best_of);\r
+        assertEquals(5, params.greedy.best_of);\r
     }\r
 \r
     @Test\r