Merge d3aca6a9de into fc674574ca
This commit is contained in:
commit
09f4775fc4
|
|
@ -8039,11 +8039,25 @@ whisper_token whisper_full_get_token_id(struct whisper_context * ctx, int i_segm
|
||||||
}
|
}
|
||||||
|
|
||||||
struct whisper_token_data whisper_full_get_token_data_from_state(struct whisper_state * state, int i_segment, int i_token) {
|
struct whisper_token_data whisper_full_get_token_data_from_state(struct whisper_state * state, int i_segment, int i_token) {
|
||||||
return state->result_all[i_segment].tokens[i_token];
|
whisper_token_data token = state->result_all[i_segment].tokens[i_token];
|
||||||
|
|
||||||
|
// Map VAD-processed token timestamps back to the original audio timeline.
|
||||||
|
// Without this, tokens report timestamps in the VAD-stripped timeline
|
||||||
|
// (starting at 0 after whisper_full_with_state sees only the speech-
|
||||||
|
// filtered samples), while segment timestamps are already mapped back by
|
||||||
|
// whisper_full_get_segment_t0/t1_from_state. The two diverge when VAD
|
||||||
|
// strips a non-speech prefix (e.g. music before speech).
|
||||||
|
// ref: https://github.com/ggml-org/whisper.cpp/issues/3754
|
||||||
|
if (state->has_vad_segments && !state->vad_mapping_table.empty()) {
|
||||||
|
token.t0 = map_processed_to_original_time(token.t0, state->vad_mapping_table);
|
||||||
|
token.t1 = map_processed_to_original_time(token.t1, state->vad_mapping_table);
|
||||||
|
}
|
||||||
|
|
||||||
|
return token;
|
||||||
}
|
}
|
||||||
|
|
||||||
struct whisper_token_data whisper_full_get_token_data(struct whisper_context * ctx, int i_segment, int i_token) {
|
struct whisper_token_data whisper_full_get_token_data(struct whisper_context * ctx, int i_segment, int i_token) {
|
||||||
return ctx->state->result_all[i_segment].tokens[i_token];
|
return whisper_full_get_token_data_from_state(ctx->state, i_segment, i_token);
|
||||||
}
|
}
|
||||||
|
|
||||||
float whisper_full_get_token_p_from_state(struct whisper_state * state, int i_segment, int i_token) {
|
float whisper_full_get_token_p_from_state(struct whisper_state * state, int i_segment, int i_token) {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue