From 530125dca86b6cecb853ce08f2dfaab461fa762a Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期四, 25 七月 2024 15:39:41 +0800
Subject: [PATCH] bug fix
---
runtime/python/libtorch/funasr_torch/sensevoice_bin.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/runtime/python/libtorch/funasr_torch/sensevoice_bin.py b/runtime/python/libtorch/funasr_torch/sensevoice_bin.py
index f103fea..8de1c54 100644
--- a/runtime/python/libtorch/funasr_torch/sensevoice_bin.py
+++ b/runtime/python/libtorch/funasr_torch/sensevoice_bin.py
@@ -163,12 +163,12 @@
_language_list = _language_list * B
if len(_textnorm_list) == 1 and B != 1:
_textnorm_list = _textnorm_list * B
-
+
ctc_logits, encoder_out_lens = self.ort_infer(
torch.Tensor(feats).to(self.device),
torch.Tensor(feats_len).to(self.device),
- torch.tensor([_language_list]).to(self.device),
- torch.tensor([_textnorm_list]).to(self.device),
+ torch.tensor(_language_list).to(self.device),
+ torch.tensor(_textnorm_list).to(self.device),
)
# support batch_size=1 only currently
x = ctc_logits[0, : encoder_out_lens[0].item(), :]
--
Gitblit v1.9.1