diff --git a/models/convert-whisper-to-coreml.py b/models/convert-whisper-to-coreml.py index f66683e9..441efdd2 100644 --- a/models/convert-whisper-to-coreml.py +++ b/models/convert-whisper-to-coreml.py @@ -245,7 +245,7 @@ def convert_encoder(hparams, model, quantize=False): model = ct.convert( traced_model, - convert_to=None if quantize else "mlprogram", # convert will fail if weights are quantized, not sure why + convert_to="neuralnetwork", inputs=[ct.TensorType(name="logmel_data", shape=input_shape)], outputs=[ct.TensorType(name="output")], compute_units=ct.ComputeUnit.ALL @@ -268,7 +268,7 @@ def convert_decoder(hparams, model, quantize=False): model = ct.convert( traced_model, - convert_to=None if quantize else "mlprogram", # convert will fail if weights are quantized, not sure why + convert_to="neuralnetwork", inputs=[ ct.TensorType(name="token_data", shape=tokens_shape, dtype=int), ct.TensorType(name="audio_data", shape=audio_shape)