From a0d60ec3b3b126ad59fe4cee4990fc90a902a0a0 Mon Sep 17 00:00:00 2001 From: dm Date: Tue, 14 Apr 2026 21:23:52 +0800 Subject: [PATCH] Fix ov_infer.py for OpenVINO 2026.1.0 compatibility - Add comment for openvino_tokenizers import requirement - Fix infer input format: dict to list for OpenVINO 2026.1.0 compatibility These changes are necessary for the model to work with OpenVINO 2026.1.0 runtime. The openvino_tokenizers import registers SentencepieceDetokenizer operation, and the list format is required by OpenVINO 2026.1.0's infer method. Co-Authored-By: Claude Opus 4.6 --- ov_infer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ov_infer.py b/ov_infer.py index df37689..967706a 100644 --- a/ov_infer.py +++ b/ov_infer.py @@ -12,7 +12,7 @@ import numpy as np from pathlib import Path from typing import List, Union, Tuple -import openvino_tokenizers +import openvino_tokenizers # Required to register SentencepieceDetokenizer operation for OpenVINO from funasr_onnx.utils.utils import ( # CharTokenizer, @@ -244,7 +244,7 @@ def infer( language: np.ndarray, textnorm: np.ndarray, ) -> Tuple[np.ndarray, np.ndarray]: - self.infer_reqeust.infer({"speech":feats, "speech_lengths":feats_len, "language":language,"textnorm":textnorm}) + self.infer_reqeust.infer([feats, feats_len, language, textnorm]) ctc_logits = self.infer_reqeust.get_tensor('ctc_logits').data.copy() encoder_out_lens = self.infer_reqeust.get_tensor('encoder_out_lens').data.copy() return ctc_logits, encoder_out_lens