mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2024-12-28 17:49:05 +01:00
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>
This commit is contained in:
parent
37947203e6
commit
ec7a6f04f9
@ -2027,7 +2027,7 @@ static bool whisper_encode_internal(
|
|||||||
wstate.t_encode_us += ggml_time_us() - t_start_us;
|
wstate.t_encode_us += ggml_time_us() - t_start_us;
|
||||||
wstate.n_encode++;
|
wstate.n_encode++;
|
||||||
|
|
||||||
return true;
|
return !(abort_callback && abort_callback(abort_callback_data));
|
||||||
}
|
}
|
||||||
|
|
||||||
static struct ggml_cgraph * whisper_build_graph_decoder(
|
static struct ggml_cgraph * whisper_build_graph_decoder(
|
||||||
@ -2447,7 +2447,7 @@ static bool whisper_decode_internal(
|
|||||||
wstate.n_prompt++;
|
wstate.n_prompt++;
|
||||||
}
|
}
|
||||||
|
|
||||||
return true;
|
return !(abort_callback && abort_callback(abort_callback_data));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user