parakeet : fix TDT decode by outputting raw logits from the joint graph (#4017)

This commit is contained in:
Jhen-Jie Hong
2026-08-29 05:28:05 +02:00
committed by GitHub
parent 978113305b
commit c4ac0012a8
2 changed files with 15 additions and 13 deletions
+2 -1
View File
@@ -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);