whisper : return with error from whisper_encode_internal and whisper_decode_internal when abort callback is true (#1456)

Co-authored-by: Ben Nortier <ben@bjnortier.com>
pull/1475/head
Ben Nortier 2023-11-10 13:51:16 +02:00 committed by GitHub
parent 37947203e6
commit ec7a6f04f9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -2027,7 +2027,7 @@ static bool whisper_encode_internal(
wstate.t_encode_us += ggml_time_us() - t_start_us;
wstate.n_encode++;
return true;
return !(abort_callback && abort_callback(abort_callback_data));
}
static struct ggml_cgraph * whisper_build_graph_decoder(
@ -2447,7 +2447,7 @@ static bool whisper_decode_internal(
wstate.n_prompt++;
}
return true;
return !(abort_callback && abort_callback(abort_callback_data));
}