From 22e51ec95fcba72e0a8016775b31ed9c60da3399 Mon Sep 17 00:00:00 2001
From: R1ckShi <shixian.shi@alibaba-inc.com>
Date: 星期一, 03 六月 2024 15:56:39 +0800
Subject: [PATCH] bug fix
---
runtime/python/libtorch/funasr_torch/paraformer_bin.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/runtime/python/libtorch/funasr_torch/paraformer_bin.py b/runtime/python/libtorch/funasr_torch/paraformer_bin.py
index 755237e..5fa3cc9 100644
--- a/runtime/python/libtorch/funasr_torch/paraformer_bin.py
+++ b/runtime/python/libtorch/funasr_torch/paraformer_bin.py
@@ -337,7 +337,7 @@
outputs = self.bb_infer(feats, feats_len, bias_embed)
am_scores, valid_token_lens = outputs[0], outputs[1]
else:
- outputs = self.bb_infer_infer(feats.cuda(), feats_len.cuda(), bias_embed.cuda())
+ outputs = self.bb_infer(feats.cuda(), feats_len.cuda(), bias_embed.cuda())
am_scores, valid_token_lens = outputs[0].cpu(), outputs[1].cpu()
except:
# logging.warning(traceback.format_exc())
--
Gitblit v1.9.1