coreml : fix ANE optimized encoder (#1716)

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

View File

@ -23,7 +23,7 @@ if [[ $mname == "-h5" ]]; then
echo $mpath
python3 models/convert-h5-to-coreml.py --model-name $mname --model-path $mpath --encoder-only True
else
python3 models/convert-whisper-to-coreml.py --model $mname --encoder-only True
python3 models/convert-whisper-to-coreml.py --model $mname --encoder-only True --optimize-ane True
fi
xcrun coremlc compile models/coreml-encoder-${mname}.mlpackage models/