bobqianic commited on
Commit
b9f3d1b
·
unverified ·
1 Parent(s): 0220892

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

Browse files
models/convert-whisper-to-openvino.py CHANGED
@@ -9,7 +9,7 @@ import shutil
9
  def convert_encoder(hparams, encoder, mname):
10
  encoder.eval()
11
 
12
- mel = torch.zeros((1, 80, 3000))
13
 
14
  onnx_folder=os.path.join(os.path.dirname(__file__),"onnx_encoder")
15
 
 
9
  def convert_encoder(hparams, encoder, mname):
10
  encoder.eval()
11
 
12
+ mel = torch.zeros((1, hparams.n_mels, 3000))
13
 
14
  onnx_folder=os.path.join(os.path.dirname(__file__),"onnx_encoder")
15