openvino : update convert-whisper-to-openvino.py to support v3 (#1459)

2023-11-09 10:42:39 +00:00 · 2023-11-09 10:42:39 +00:00 · 953419c69a
parent 0de8582f65
commit 953419c69a
1 changed files with 1 additions and 1 deletions
--- a/models/convert-whisper-to-openvino.py
+++ b/models/convert-whisper-to-openvino.py
@ -9,7 +9,7 @@ import shutil
 def convert_encoder(hparams, encoder, mname):
    encoder.eval()

-    mel = torch.zeros((1, 80, 3000))
+    mel = torch.zeros((1, hparams.n_mels, 3000))

    onnx_folder=os.path.join(os.path.dirname(__file__),"onnx_encoder")