From c4e70144920401a4f5d9b569929b6914af5d324c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 24 一月 2025 16:59:23 +0800
Subject: [PATCH] v1.2.3
---
runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py b/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py
index 9cfe46f..6a06ed1 100644
--- a/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py
+++ b/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py
@@ -183,8 +183,10 @@
# ctc_logits = torch.from_numpy(ctc_logits).float()
# support batch_size=1 only currently
x = ctc_logits[b, : encoder_out_lens[b].item(), :]
- yseq = x.argmax(dim=-1)
- yseq = np.unique(yseq)
+ yseq = np.argmax(x, axis=-1)
+ # Use np.diff and np.where instead of torch.unique_consecutive.
+ mask = np.concatenate(([True], np.diff(yseq) != 0))
+ yseq = yseq[mask]
mask = yseq != self.blank_id
token_int = yseq[mask].tolist()
--
Gitblit v1.9.1