mgrachten commited on
Commit
e351248
·
unverified ·
1 Parent(s): b349a86

coreml : always convert to "neuralnetwork" (#2770)

Browse files
models/convert-whisper-to-coreml.py CHANGED
@@ -245,7 +245,7 @@ def convert_encoder(hparams, model, quantize=False):
245
 
246
  model = ct.convert(
247
  traced_model,
248
- convert_to=None if quantize else "mlprogram", # convert will fail if weights are quantized, not sure why
249
  inputs=[ct.TensorType(name="logmel_data", shape=input_shape)],
250
  outputs=[ct.TensorType(name="output")],
251
  compute_units=ct.ComputeUnit.ALL
@@ -268,7 +268,7 @@ def convert_decoder(hparams, model, quantize=False):
268
 
269
  model = ct.convert(
270
  traced_model,
271
- convert_to=None if quantize else "mlprogram", # convert will fail if weights are quantized, not sure why
272
  inputs=[
273
  ct.TensorType(name="token_data", shape=tokens_shape, dtype=int),
274
  ct.TensorType(name="audio_data", shape=audio_shape)
 
245
 
246
  model = ct.convert(
247
  traced_model,
248
+ convert_to="neuralnetwork",
249
  inputs=[ct.TensorType(name="logmel_data", shape=input_shape)],
250
  outputs=[ct.TensorType(name="output")],
251
  compute_units=ct.ComputeUnit.ALL
 
268
 
269
  model = ct.convert(
270
  traced_model,
271
+ convert_to="neuralnetwork",
272
  inputs=[
273
  ct.TensorType(name="token_data", shape=tokens_shape, dtype=int),
274
  ct.TensorType(name="audio_data", shape=audio_shape)