Spaces:
Running
Running
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=
|
| 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=
|
| 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)
|