Spaces:
Sleeping
Sleeping
bobqianic
commited on
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,
|
| 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 |
|