From 0d36a5e176224123c826a254e80c61b7b6dc435a Mon Sep 17 00:00:00 2001
From: wu0792 <wu0792@users.noreply.github.com>
Date: 星期四, 25 七月 2024 18:55:29 +0800
Subject: [PATCH] Fix Typo (#1960)
---
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