mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-01-27 14:49:55 +00:00
whisper : fix CoreML
This commit is contained in:
parent
0ab5025316
commit
a54d8c9dec
@ -1939,7 +1939,12 @@ static struct ggml_cgraph * whisper_build_graph_cross(
|
||||
|
||||
ggml_allocr * alloc = wstate.alloc_cross.alloc;
|
||||
|
||||
struct ggml_tensor * cur = ggml_view_tensor(ctx0, wstate.embd_enc);
|
||||
struct ggml_tensor * cur = ggml_new_tensor_2d(ctx0, GGML_TYPE_F32, n_state, n_ctx);
|
||||
ggml_allocr_alloc(alloc, cur);
|
||||
|
||||
if (!ggml_allocr_is_measure(alloc)) {
|
||||
ggml_backend_tensor_copy(wstate.embd_enc, cur);
|
||||
}
|
||||
|
||||
struct ggml_tensor * Kscale = ggml_new_tensor_1d(ctx0, GGML_TYPE_F32, 1);
|
||||
ggml_allocr_alloc(alloc, Kscale);
|
||||
|
Loading…
x
Reference in New Issue
Block a user