Fix whisper test script for the latest onnxruntime. (#494)

2023-12-20 11:12:12 +08:00
parent 03ff9db56e
commit ef8d112aaa
3 changed files with 18 additions and 3 deletions
--- a/scripts/whisper/test.py
+++ b/scripts/whisper/test.py
@@ -82,6 +82,7 @@ class OnnxModel:
        self.encoder = ort.InferenceSession(
            encoder,
            sess_options=self.session_opts,
+            providers=["CPUExecutionProvider"],
        )

        meta = self.encoder.get_modelmeta().custom_metadata_map
@@ -113,6 +114,7 @@ class OnnxModel:
        self.decoder = ort.InferenceSession(
            decoder,
            sess_options=self.session_opts,
+            providers=["CPUExecutionProvider"],
        )

    def run_encoder(