extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-08-18 01:00:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

coreml : fix ANE optimized encoder (#1716)

Browse Source
This commit is contained in:
Yajing Tang
2024-01-04 06:28:30 -08:00
committed by GitHub
parent ab0a8593c5
commit ba5bcde874
3 changed files with 4 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
coreml/whisper-encoder.mm
View File

@@ -24,9 +24,9 @@ struct whisper_coreml_context * whisper_coreml_init(const char * path_model) {
// select which device to run the Core ML model on
MLModelConfiguration *config = [[MLModelConfiguration alloc] init];
config.computeUnits = MLComputeUnitsCPUAndGPU;
// config.computeUnits = MLComputeUnitsCPUAndGPU;
//config.computeUnits = MLComputeUnitsCPUAndNeuralEngine;
//config.computeUnits = MLComputeUnitsAll;
config.computeUnits = MLComputeUnitsAll;
const void * data = CFBridgingRetain([[whisper_encoder_impl alloc] initWithContentsOfURL:url_model configuration:config error:nil]);