openvino : update convert-whisper-to-openvino.py to support v3 (ggerg…

…anov#1459)
felrock · Nov 18, 2023 · 724aae7 · 724aae7
1 parent 597b433
commit 724aae7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/convert-whisper-to-openvino.py b/models/convert-whisper-to-openvino.py
@@ -9,7 +9,7 @@
 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")