Skip to content

Commit 2c54643

Browse files
chore: fix whisper evaluate (microsoft#226)
Co-authored-by: ziyuanguo <ziyuanguo@microsoft.com>
1 parent 0b6d52a commit 2c54643

1 file changed

Lines changed: 15 additions & 0 deletions

File tree

openai-whisper-large-v3-turbo/aitk/qnn_evaluate.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,20 @@
1010
logger = logging.getLogger(os.path.basename(__file__))
1111
logging.basicConfig(level=logging.INFO)
1212

13+
def register_execution_providers():
14+
import subprocess
15+
import sys
16+
import onnxruntime as ort
17+
18+
worker_script = os.path.abspath('winml.py')
19+
result = subprocess.check_output([sys.executable, worker_script], text=True)
20+
paths = json.loads(result)
21+
for item in paths.items():
22+
try:
23+
ort.register_execution_provider_library(item[0], item[1])
24+
except Exception as e:
25+
logger.warning(f"Failed to register execution provider {item[0]}: {e}")
26+
1327
def main():
1428
parser = argparse.ArgumentParser(description="Evaluate Whisper")
1529
parser.add_argument(
@@ -56,6 +70,7 @@ def main():
5670
encoder_path = args.encoder
5771
decoder_path = args.decoder
5872

73+
register_execution_providers()
5974
app = HfWhisperAppWithSave(encoder_path, decoder_path, args.model_id, args.execution_provider, get_device_type(args.device_str))
6075

6176
encoder_latencies = []

0 commit comments

Comments
 (0)