Spaces:
Sleeping
Sleeping
bobqianic
commited on
models : use absolute paths for the converted model (#1356)
Browse files
models/convert-whisper-to-openvino.py
CHANGED
|
@@ -29,7 +29,7 @@ def convert_encoder(hparams, encoder, mname):
|
|
| 29 |
|
| 30 |
# use model optimizer to convert onnx to OpenVINO IR format
|
| 31 |
encoder_model = mo.convert_model(onnx_path, compress_to_fp16=True)
|
| 32 |
-
serialize(encoder_model, xml_path=
|
| 33 |
|
| 34 |
#cleanup
|
| 35 |
if os.path.isdir(onnx_folder):
|
|
|
|
| 29 |
|
| 30 |
# use model optimizer to convert onnx to OpenVINO IR format
|
| 31 |
encoder_model = mo.convert_model(onnx_path, compress_to_fp16=True)
|
| 32 |
+
serialize(encoder_model, xml_path=os.path.join(os.path.dirname(__file__),"ggml-" + mname + "-encoder-openvino.xml"))
|
| 33 |
|
| 34 |
#cleanup
|
| 35 |
if os.path.isdir(onnx_folder):
|