diff --git a/models/convert-whisper-to-coreml.py b/models/convert-whisper-to-coreml.py index 3876a287..66827b6d 100644 --- a/models/convert-whisper-to-coreml.py +++ b/models/convert-whisper-to-coreml.py @@ -254,10 +254,10 @@ def convert_encoder(hparams, model, quantize=False): model = ct.convert( traced_model, - convert_to="neuralnetwork", + convert_to="mlprogram", inputs=[ct.TensorType(name="logmel_data", shape=input_shape)], outputs=[ct.TensorType(name="output")], - compute_units=ct.ComputeUnit.ALL + compute_units=ct.ComputeUnit.ALL, ) if quantize: @@ -278,11 +278,11 @@ def convert_decoder(hparams, model, quantize=False): model = ct.convert( traced_model, - convert_to="neuralnetwork", + convert_to="mlprogram", inputs=[ ct.TensorType(name="token_data", shape=tokens_shape, dtype=int), ct.TensorType(name="audio_data", shape=audio_shape) - ] + ], ) if quantize: