ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-12-20 05:07:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
bobqianic 2023-11-09 10:42:39 +00:00 committed by GitHub
parent 0de8582f65
commit 953419c69a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/convert-whisper-to-openvino.py
View File

@ -9,7 +9,7 @@ import shutil
def convert_encoder(hparams, encoder, mname): def convert_encoder(hparams, encoder, mname):
encoder.eval() 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") onnx_folder=os.path.join(os.path.dirname(__file__),"onnx_encoder")