mirror of
https://github.com/ggml-org/whisper.cpp.git
synced 2026-09-30 11:36:38 +02:00
parakeet : fix TDT decode by outputting raw logits from the joint graph (#4017)
This commit is contained in:
+2
-1
@@ -176,7 +176,8 @@ extern "C" {
|
||||
// Token logits obtained from the last call to parakeet_full/parakeet_chunk
|
||||
// The logits for the last token are stored in the last row
|
||||
// Rows: n_tokens
|
||||
// Cols: n_vocab
|
||||
// Cols: n_vocab + 1 token logits (the blank token is at index n_vocab),
|
||||
// followed by n_tdt_durations duration logits
|
||||
PARAKEET_API float * parakeet_get_logits (struct parakeet_context * ctx);
|
||||
PARAKEET_API float * parakeet_get_logits_from_state(struct parakeet_state * state);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user