From 1d79e78402abaa9d92e1f55bb15316443b29ea9f Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Wed, 15 Nov 2023 17:42:53 +0200 Subject: [PATCH] java : fix test (#1492) --- .../java/io/github/ggerganov/whispercpp/WhisperCppTest.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bindings/java/src/test/java/io/github/ggerganov/whispercpp/WhisperCppTest.java b/bindings/java/src/test/java/io/github/ggerganov/whispercpp/WhisperCppTest.java index 2973243d..034726ad 100644 --- a/bindings/java/src/test/java/io/github/ggerganov/whispercpp/WhisperCppTest.java +++ b/bindings/java/src/test/java/io/github/ggerganov/whispercpp/WhisperCppTest.java @@ -45,7 +45,7 @@ class WhisperCppTest { assertEquals(16384, params.n_max_text_ctx); assertFalse(params.translate); assertEquals(0.01f, params.thold_pt); - assertEquals(2, params.beam_search.beam_size); + assertEquals(5, params.beam_search.beam_size); assertEquals(-1.0f, params.beam_search.patience); } @@ -58,7 +58,7 @@ class WhisperCppTest { assertEquals(WhisperSamplingStrategy.WHISPER_SAMPLING_GREEDY.ordinal(), params.strategy); assertNotEquals(0, params.n_threads); assertEquals(16384, params.n_max_text_ctx); - assertEquals(2, params.greedy.best_of); + assertEquals(5, params.greedy.best_of); } @Test