ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-01-29 15:44:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

whisper : fix with_state methods to use the correct state (#1519)

Browse Source 
Co-authored-by: Sandro Hanea <sandrohanea@microsoft.com>
This commit is contained in:
sandrohanea 2023-11-19 10:25:30 +01:00 committed by GitHub
parent f784f9fa12
commit 46cc26d1b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
whisper.cpp
View File

@ -3525,7 +3525,7 @@ int whisper_encode(struct whisper_context * ctx, int offset, int n_threads) {
int whisper_decode_with_state(struct whisper_context * ctx, struct whisper_state * state, const whisper_token * tokens, int n_tokens, int n_past, int n_threads) {
whisper_batch_prep_legacy(state->batch, tokens, n_tokens, n_past, 0);
whisper_kv_cache_seq_rm(ctx->state->kv_self, 0, n_past, -1);
whisper_kv_cache_seq_rm(state->kv_self, 0, n_past, -1);
if (!whisper_decode_internal(*ctx, *state, state->batch, n_threads, nullptr, nullptr)) {
WHISPER_LOG_ERROR("%s: failed to eval\n", __func__);
@ -5186,7 +5186,7 @@ int whisper_full_with_state(
const int progress_cur = (100*(seek - seek_start))/(seek_end - seek_start);
params.progress_callback(
ctx, ctx->state, progress_cur, params.progress_callback_user_data);
ctx, state, progress_cur, params.progress_callback_user_data);
}
// of only 1 second left, then stop