]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
openvino : update convert-whisper-to-openvino.py to support v3 (#1459)
authorbobqianic <redacted>
Thu, 9 Nov 2023 10:42:39 +0000 (10:42 +0000)
committerGitHub <redacted>
Thu, 9 Nov 2023 10:42:39 +0000 (12:42 +0200)
models/convert-whisper-to-openvino.py

index 6b3d396643bad8affd56dcd98c1f8cb885d5a0d5..88e03ff7550541b200a647735c5e0af960820f54 100644 (file)
@@ -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")