This commit is contained in:
AlanConstantino 2026-04-20 15:30:44 +00:00 committed by GitHub
commit d7b9bdf679
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions

View File

@ -25,7 +25,10 @@ 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 --optimize-ane True
# Note: --optimize-ane is disabled as it produces models that crash during inference.
# The ANE optimization code needs maintenance to work with newer model architectures
# and coremltools versions. See convert-whisper-to-coreml.py for details.
python3 models/convert-whisper-to-coreml.py --model "$mname" --encoder-only True
fi
xcrun coremlc compile models/coreml-encoder-"${mname}".mlpackage models/