From 953419c69ab4825954157144caba2baeecfdb9aa Mon Sep 17 00:00:00 2001 From: bobqianic <129547291+bobqianic@users.noreply.github.com> Date: Thu, 9 Nov 2023 10:42:39 +0000 Subject: [PATCH] openvino : update convert-whisper-to-openvino.py to support v3 (#1459) --- models/convert-whisper-to-openvino.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/convert-whisper-to-openvino.py b/models/convert-whisper-to-openvino.py index 6b3d3966..88e03ff7 100644 --- 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")