close #113 : fix struct whisper_token_data

2 years ago · eba62e0fa1
parent 69bdb6624a
commit eba62e0fa1
2 changed files with 2 additions and 2 deletions
--- a/whisper.cpp
+++ b/whisper.cpp
@ -2209,7 +2209,7 @@ int whisper_decode(struct whisper_context * ctx, const whisper_token * tokens, i
    return 0;
 }

-whisper_token_data whisper_sample_best(struct whisper_context * ctx) {
+struct whisper_token_data whisper_sample_best(struct whisper_context * ctx) {
    const int64_t t_start_sample_us = ggml_time_us();

    // TODO: simplify
--- a/whisper.h
+++ b/whisper.h
@ -128,7 +128,7 @@ extern "C" {
    // You can also implement your own sampling method using the whisper_get_probs() function.
    // whisper_sample_best() returns the token with the highest probability
    // whisper_sample_timestamp() returns the most probable timestamp token
-    WHISPER_API whisper_token_data whisper_sample_best(struct whisper_context * ctx);
+    WHISPER_API struct whisper_token_data whisper_sample_best(struct whisper_context * ctx);
    WHISPER_API whisper_token whisper_sample_timestamp(struct whisper_context * ctx);

    // Return the id of the specified language, returns -1 if not found