Merge 9acac86ec4 into fc674574ca
This commit is contained in:
commit
d7b9bdf679
|
|
@ -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/
|
||||
|
|
|
|||
Loading…
Reference in New Issue