coreml : simlpify whisper_encode + log messages

coreml
Georgi Gerganov 2 years ago
parent b0ac915265
commit 17a14593de
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

@ -1405,6 +1405,7 @@ static bool whisper_encode(
}
}
#ifndef WHISPER_USE_COREML
struct ggml_tensor * cur;
// convolution + gelu
@ -1704,9 +1705,6 @@ static bool whisper_encode(
wctx.use_buf(ctx0, -1);
// run the computation
#ifdef WHISPER_USE_COREML
whisper_coreml_encode(wctx.ctx_coreml, (float *) mel->data, (float *) cur->data);
#else
{
struct ggml_cgraph gf = {};
gf.n_threads = n_threads;
@ -1716,6 +1714,12 @@ static bool whisper_encode(
//ggml_graph_print(&gf);
}
#else
wctx.use_buf(ctx0, -1);
struct ggml_tensor * cur = ggml_new_tensor_2d(ctx0, GGML_TYPE_F32, n_state, n_ctx);
whisper_coreml_encode(wctx.ctx_coreml, (float *) mel->data, (float *) cur->data);
#endif
// cur
@ -2569,12 +2573,15 @@ struct whisper_context * whisper_init_from_file(const char * path_model) {
#ifdef WHISPER_USE_COREML
const auto path_coreml = whisper_get_coreml_path(ctx->path_model);
fprintf(stderr, "%s: loading Core ML model from '%s'\n", __func__, path_coreml.c_str());
fprintf(stderr, "%s: first run on a device may take a while ...\n", __func__);
ctx->ctx_coreml = whisper_coreml_init(path_coreml.c_str());
if (!ctx->ctx_coreml) {
fprintf(stderr, "%s: failed to load Core ML model from '%s'\n", __func__, path_coreml.c_str());
return nullptr;
}
fprintf(stderr, "%s: Core ML model loaded\n", __func__);
#endif
}

Loading…
Cancel
Save